Merge pull request #76 from ponder-lab/fix_tests_again

Move the value number check outside the context loop.

Author: khatchad

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

@@ -328,6 +328,12 @@ private void testTf2(
 
     assertEquals(expectedNumberOfTensorVariables, functionTensorVariables.size());
 
+    // check value numbers.
+    assertEquals(
+        "Each tensor parameter should have a unique value number.",
+        expectedNumberOfTensorParameters,
+        expectedTensorParameterValueNumbers.length);
+
     // get the pointer keys for the function by their contexts.
     Map<Context, Set<LocalPointerKey>> contextToFunctionParameterPointerKeys =
         functionSignatureToPointerKeys.getOrDefault(functionSignature, emptySet()).stream()
@@ -342,11 +348,6 @@ 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)