From bfe02461e071af5ad7240283cadcf43a5443407a Mon Sep 17 00:00:00 2001 From: Raffi Khatchadourian Date: Wed, 11 Oct 2023 11:22:23 -0400 Subject: [PATCH] Add test case for https://github.com/wala/ML/issues/89. --- .../wala/cast/python/ml/test/TestTensorflowModel.java | 6 ++++++ .../data/tf2_test_dataset.py | 11 +++++++++++ 2 files changed, 17 insertions(+) create mode 100644 com.ibm.wala.cast.python.test/data/tf2_test_dataset.py diff --git a/com.ibm.wala.cast.python.ml.test/source/com/ibm/wala/cast/python/ml/test/TestTensorflowModel.java b/com.ibm.wala.cast.python.ml.test/source/com/ibm/wala/cast/python/ml/test/TestTensorflowModel.java index 0ac722db8..4e72744ce 100644 --- a/com.ibm.wala.cast.python.ml.test/source/com/ibm/wala/cast/python/ml/test/TestTensorflowModel.java +++ b/com.ibm.wala.cast.python.ml.test/source/com/ibm/wala/cast/python/ml/test/TestTensorflowModel.java @@ -195,6 +195,12 @@ public void testTf2() testTf2("tf2_testing_decorator8.py", "returned", 1, 3, 2); testTf2("tf2_testing_decorator9.py", "returned", 1, 3, 2); testTf2("tf2_testing_decorator10.py", "returned", 1, 3, 2); + testTf2( + "tf2_test_dataset.py", + "add", + 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. } private void testTf2( diff --git a/com.ibm.wala.cast.python.test/data/tf2_test_dataset.py b/com.ibm.wala.cast.python.test/data/tf2_test_dataset.py new file mode 100644 index 000000000..51d8a4b77 --- /dev/null +++ b/com.ibm.wala.cast.python.test/data/tf2_test_dataset.py @@ -0,0 +1,11 @@ +import tensorflow as tf + + +def add(a, b): + return a + b + + +dataset = tf.data.Dataset.from_tensor_slices([1, 2, 3]) + +for element in dataset: + c = add(element, element)