Add test for simple class method

Author: tamarinvs19

utbot-python/src/test/java/org/utbot/python/framework/external/PythonUtBotJavaApiTest.java

@@ -28,7 +28,7 @@ public void setUp() {
     public void tearDown() {
         Cleaner.INSTANCE.doCleaning();
     }
-    private File loadExampleCode(String name) {
+    private File loadResource(String name) {
         URL resource = getClass().getClassLoader().getResource(name);
         if (resource == null) {
             throw new IllegalArgumentException("file not found!");
@@ -43,7 +43,7 @@ private File loadExampleCode(String name) {
 
     @Test
     public void testSimpleFunction() {
-        File fileWithCode = loadExampleCode("example_code/arithmetic.py");
+        File fileWithCode = loadResource("example_code/arithmetic.py");
         String pythonRunRoot = fileWithCode.getParentFile().getAbsolutePath();
         String moduleFilename = fileWithCode.getAbsolutePath();
         PythonObjectName testMethodName = new PythonObjectName("arithmetic", "calculate_function_value");
@@ -66,7 +66,7 @@ public void testSimpleFunction() {
 
     @Test
     public void testSimpleFunctionTestCase() {
-        File fileWithCode = loadExampleCode("example_code/arithmetic.py");
+        File fileWithCode = loadResource("example_code/arithmetic.py");
         String pythonRunRoot = fileWithCode.getParentFile().getAbsolutePath();
         String moduleFilename = fileWithCode.getAbsolutePath();
         PythonObjectName testMethodName = new PythonObjectName("arithmetic", "calculate_function_value");
@@ -86,4 +86,29 @@ public void testSimpleFunctionTestCase() {
         Assertions.assertTrue(testCase.size() > 0);
         Assertions.assertTrue(testCase.get(0).component2().size() > 0);
     }
+
+    @Test
+    public void testSimpleClassMethod() {
+        File fileWithCode = loadResource("example_code/inner_dir/inner_module.py");
+        File projectRoot = loadResource("example_code/");
+        String pythonRunRoot = projectRoot.getAbsolutePath();
+        String moduleFilename = fileWithCode.getAbsolutePath();
+        PythonObjectName containingClassName = new PythonObjectName("inner_dir.inner_module", "InnerClass");
+        PythonObjectName testMethodName = new PythonObjectName("inner_dir.inner_module", "f");
+        PythonTestMethodInfo methodInfo = new PythonTestMethodInfo(testMethodName, moduleFilename, containingClassName);
+        ArrayList<PythonTestMethodInfo> testMethods = new ArrayList<>(1);
+        testMethods.add(methodInfo);
+        ArrayList<String> directoriesForSysPath = new ArrayList<>();
+        directoriesForSysPath.add(pythonRunRoot);
+        String testCode = PythonUtBotJavaApi.generate(
+                testMethods,
+                pythonPath,
+                pythonRunRoot,
+                directoriesForSysPath,
+                10_000,
+                1_000,
+                Unittest.INSTANCE
+        );
+        Assertions.assertTrue(testCode.length() > 0);
+    }
 }

utbot-python/src/test/resources/example_code/__init__.py

@@ -0,0 +1,8 @@
+class InnerClass:
+    x: int
+
+    def __init__(self, x: int):
+        self.x = x
+
+    def f(self, y: int):
+        return y**2 + self.x*y + 1