Add new list test.

Just numeric values here as a "control."
ponder-lab · Dec 15, 2023 · e20e5a6 · e20e5a6
1 parent e7dc30d
commit e20e5a6
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
@@ -208,6 +208,7 @@ public void testTf2()
     testTf2("tf2_test_tensor_list.py", "add", 2, 3, 2, 3);
     testTf2("tf2_test_tensor_list2.py", "add", 0, 2);
     testTf2("tf2_test_tensor_list3.py", "add", 0, 2);
+    testTf2("tf2_test_tensor_list4.py", "add", 0, 0);
     testTf2("tf2_test_model_call.py", "SequentialModel.__call__", 1, 4, 3);
     testTf2("tf2_test_model_call2.py", "SequentialModel.call", 1, 4, 3);
     testTf2("tf2_test_model_call3.py", "SequentialModel.call", 1, 4, 3);

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