Check the context/funciton PK mappings. (#77)

* Check the context/function PK mappings. We need at least one if we are expecting at least one tensor parameter.
ponder-lab · Jan 23, 2024 · eb4f301 · eb4f301
1 parent 663d27b
commit eb4f301
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
@@ -348,6 +348,11 @@ private void testTf2(
             .filter(LocalPointerKey::isParameter)
             .collect(groupingBy(lpk -> lpk.getNode().getContext(), toSet()));
 
+    assertTrue(
+        "Because tensor parameters are inferred via function arguments, we need at least one"
+            + " calling context if we are expecting at least one tensor parameter.",
+        expectedNumberOfTensorParameters <= 0 || contextToFunctionParameterPointerKeys.size() > 0);
+
     for (Context ctx : contextToFunctionParameterPointerKeys.keySet()) {
       // check tensor parameters.
       Set<LocalPointerKey> functionParameterPointerKeys =