Add more shape checks.

Author: khatchad

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

@@ -873,11 +873,15 @@ public void testAdd7()
     Dimension<Integer> aX = new NumericDim(1);
     Dimension<Integer> aY = new NumericDim(2);
 
+    List<Dimension<?>> aDimensions = asList(aX, aY);
+
     Dimension<Integer> bX = new NumericDim(2);
     Dimension<Integer> bY = new NumericDim(2);
 
-    TensorType expectedTypeForA = new TensorType("pixel", asList(aX, aY));
-    TensorType expectedTypeForB = new TensorType("pixel", asList(bX, bY));
+    List<Dimension<?>> bDimensions = asList(bX, bY);
+
+    TensorType expectedTypeForA = new TensorType("pixel", aDimensions);
+    TensorType expectedTypeForB = new TensorType("pixel", bDimensions);
 
     test(
         "tf2_test_add7.py",