More tests for wala#195.

Author: khatchad

com.ibm.wala.cast.python.ml.test/source/com/ibm/wala/cast/python/ml/test/TestTensorflow2Model.java

@@ -3361,6 +3361,18 @@ public void testReshape3() throws ClassHierarchyException, CancelException, IOEx
     test("tf2_test_reshape3.py", "f", 1, 1, 2);
   }
 
+  /** Test https://github.com/wala/ML/issues/195. */
+  @Test
+  public void testReshape4() throws ClassHierarchyException, CancelException, IOException {
+    test("tf2_test_reshape4.py", "f", 1, 1, 2);
+  }
+
+  /** Test https://github.com/wala/ML/issues/195. */
+  @Test
+  public void testReshape5() throws ClassHierarchyException, CancelException, IOException {
+    test("tf2_test_reshape.py", "f", 1, 1, 2);
+  }
+
   private void test(
       String filename,
       String functionName,

com.ibm.wala.cast.python.test/data/tf2_test_reshape4.py

@@ -0,0 +1,12 @@
+# https://www.tensorflow.org/versions/r2.9/api_docs/python/tf/reshape
+
+import tensorflow as tf
+
+
+def f(a):
+    pass
+
+
+t1 = tf.ones([28, 28])
+t2 = tf.reshape(t1, [-1, 28, 28, 1])
+f(t2)

com.ibm.wala.cast.python.test/data/tf2_test_reshape5.py

@@ -0,0 +1,12 @@
+# https://www.tensorflow.org/versions/r2.9/api_docs/python/tf/reshape
+
+import tensorflow as tf
+
+
+def f(a):
+    pass
+
+
+t1 = tf.ones([2, 3])
+t2 = tf.reshape(t1, [-1, 28, 28, 1])
+f(t2)