Another test.

Another test for wala#211.
ponder-lab · Jul 22, 2024 · e273254 · e273254
1 parent 3eeebb5
commit e273254
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/...ala.cast.python.ml.test/source/com/ibm/wala/cast/python/ml/test/TestTensorflow2Model.java b/...ala.cast.python.ml.test/source/com/ibm/wala/cast/python/ml/test/TestTensorflow2Model.java
@@ -2969,6 +2969,13 @@ public void testModule77()
         new int[] {2});
   }
 
+  /** Test https://github.com/wala/ML/issues/211. */
+  @Test
+  public void testModule78()
+      throws ClassHierarchyException, IllegalArgumentException, CancelException, IOException {
+    test(new String[] {"module.py", "client2.py"}, "module.py", "f", "", 1, 1, new int[] {2});
+  }
+
   @Test
   public void testStaticMethod() throws ClassHierarchyException, CancelException, IOException {
     test("tf2_test_static_method.py", "MyClass.the_static_method", 1, 1, 2);

diff --git a/com.ibm.wala.cast.python.test/data/client2.py b/com.ibm.wala.cast.python.test/data/client2.py
@@ -0,0 +1,6 @@
+# Test https://github.com/wala/ML/issues/211.
+
+from tensorflow import ones
+from module import f
+
+f(ones([1, 2]))