Additional check.

ponder-lab · Jan 19, 2024 · 579ccd8 · 579ccd8
1 parent 13b5397
commit 579ccd8
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/...wala.cast.python.ml.test/source/com/ibm/wala/cast/python/ml/test/TestTensorflowModel.java b/...wala.cast.python.ml.test/source/com/ibm/wala/cast/python/ml/test/TestTensorflowModel.java
@@ -338,6 +338,11 @@ private void testTf2(
       assertEquals(expectedNumberOfTensorParameters, functionParameterPointerKeys.size());
 
       // check value numbers.
+      assertEquals(
+          "Each tensor parameter should have a unique value number.",
+          expectedNumberOfTensorParameters,
+          expectedTensorParameterValueNumbers.length);
+
       Set<Integer> actualParameterValueNumberSet =
           functionParameterPointerKeys.stream()
               .map(LocalPointerKey::getValueNumber)