diff --git a/com.ibm.wala.cast.python.ml.test/source/com/ibm/wala/cast/python/ml/test/TestTensorflow2Model.java b/com.ibm.wala.cast.python.ml.test/source/com/ibm/wala/cast/python/ml/test/TestTensorflow2Model.java index df0c7f89f..2133c4c84 100644 --- a/com.ibm.wala.cast.python.ml.test/source/com/ibm/wala/cast/python/ml/test/TestTensorflow2Model.java +++ b/com.ibm.wala.cast.python.ml.test/source/com/ibm/wala/cast/python/ml/test/TestTensorflow2Model.java @@ -3619,6 +3619,11 @@ public void testReshape2() throws ClassHierarchyException, CancelException, IOEx test("tf2_test_reshape2.py", "f", 1, 1, 2); } + @Test + public void testReshape3() throws ClassHierarchyException, CancelException, IOException { + test("tf2_test_reshape3.py", "f", 1, 1, 2); + } + private void test( String filename, String functionName, diff --git a/com.ibm.wala.cast.python.test/data/tf2_test_reshape3.py b/com.ibm.wala.cast.python.test/data/tf2_test_reshape3.py new file mode 100644 index 000000000..ed12f0280 --- /dev/null +++ b/com.ibm.wala.cast.python.test/data/tf2_test_reshape3.py @@ -0,0 +1,13 @@ +# https://www.tensorflow.org/versions/r2.9/api_docs/python/tf/reshape + +import tensorflow as tf + + +def f(a): + pass + + +t1 = [[1, 2, 3], [4, 5, 6]] + +t2 = tf.reshape(t1, [6]) +f(t2)