Add test for wala#202.

com.ibm.wala.cast.python.ml.test/source/com/ibm/wala/cast/python/ml/test/TestTensorflow2Model.java

@@ -2704,6 +2704,20 @@ public void testModule60()
         new int[] {2});
   }
 
+  /** Test https://github.com/wala/ML/issues/202. */
+  @Test
+  public void testModule61()
+      throws ClassHierarchyException, IllegalArgumentException, CancelException, IOException {
+    test(
+        new String[] {"proj56/src/__init__.py", "proj56/src/B.py", "proj56/A.py"},
+        "src/B.py",
+        "C.f",
+        "proj56",
+        1,
+        1,
+        new int[] {3});
+  }
+
   @Test
   public void testStaticMethod() throws ClassHierarchyException, CancelException, IOException {
     test("tf2_test_static_method.py", "MyClass.the_static_method", 1, 1, 2);

com.ibm.wala.cast.python.test/.pydevproject

@@ -19,5 +19,6 @@
     <path>/${PROJECT_DIR_NAME}/data/proj51</path>
     <path>/${PROJECT_DIR_NAME}/data/proj52</path>
     <path>/${PROJECT_DIR_NAME}/data/proj55</path>
+    <path>/${PROJECT_DIR_NAME}/data/proj56</path>
   </pydev_pathproperty>
 </pydev_project>

com.ibm.wala.cast.python.test/data/proj56/A.py

@@ -0,0 +1,13 @@
+# Test https://github.com/wala/ML/issues/163.
+
+from tensorflow import ones
+from src import C
+
+
+class D:
+
+    def __init__(self):
+        C().f(ones([1, 2]))
+
+
+D()

com.ibm.wala.cast.python.test/data/proj56/src/B.py

@@ -0,0 +1,9 @@
+# Test https://github.com/wala/ML/issues/163.
+
+from tensorflow import Tensor
+
+
+class C:
+
+    def f(self, a):
+        assert isinstance(a, Tensor)

com.ibm.wala.cast.python.test/data/proj56/src/__init__.py

@@ -0,0 +1 @@
+from .B import C