Additional test for wala#89.

Pass the entire list.
ponder-lab · Oct 11, 2023 · f627484 · f627484
1 parent e14ffb3
commit f627484
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/...wala.cast.python.ml.test/source/com/ibm/wala/cast/python/ml/test/TestTensorflowModel.java b/...wala.cast.python.ml.test/source/com/ibm/wala/cast/python/ml/test/TestTensorflowModel.java
@@ -202,6 +202,7 @@ public void testTf2()
         0); // NOTE: Change to testTf2("tf2_test_dataset.py", "add", 2, 3, 2, 3) once
     // https://github.com/wala/ML/issues/89 is fixed.
     testTf2("tf2_test_tensor_list.py", "add", 2, 3, 2, 3);
+    testTf2("tf2_test_tensor_list2.py", "add", 0, 2);
   }
 
   private void testTf2(

diff --git a/com.ibm.wala.cast.python.test/data/tf2_test_tensor_list2.py b/com.ibm.wala.cast.python.test/data/tf2_test_tensor_list2.py
@@ -0,0 +1,10 @@
+import tensorflow as tf
+
+
+def add(a, b):
+  return a + b
+
+
+list = [tf.ones([1, 2]), tf.ones([2, 2])]
+
+c = add(list, list)