Additional test for wala#89.

Use list instead of Dataset.
ponder-lab · Oct 11, 2023 · e14ffb3 · e14ffb3
1 parent 6bc042b
commit e14ffb3
Show file tree

Hide file tree

Showing 2 changed files with 12 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
@@ -201,6 +201,7 @@ public void testTf2()
         0,
         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);
   }
 
   private void testTf2(

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