Add test.

Author: khatchad

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

@@ -4364,6 +4364,12 @@ public void testEye2()
     test("tf2_test_eye2.py", "f", 1, 1, Map.of(2, Set.of(TENSOR_2_2_FLOAT32)));
   }
 
+  @Test
+  public void testEye3()
+      throws ClassHierarchyException, IllegalArgumentException, CancelException, IOException {
+    test("tf2_test_eye3.py", "f", 1, 1, Map.of(2, Set.of(TENSOR_2_3_FLOAT32)));
+  }
+
   private void test(
       String filename,
       String functionName,

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

@@ -0,0 +1,13 @@
+import tensorflow as tf
+
+
+def f(a):
+    pass
+
+
+# Construct one 2 x 3 "identity" matrix
+arg = tf.eye(2, 3)
+assert arg.shape == (2, 3)
+assert arg.dtype == tf.float32
+
+f(arg)