Skip to content

Commit 0dd22d9

Browse files
committed
Minor cleanup.
1 parent ef10179 commit 0dd22d9

File tree

1 file changed

+9
-9
lines changed

1 file changed

+9
-9
lines changed

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

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -3398,7 +3398,7 @@ private void test(
33983398
String functionName,
33993399
String pythonPath,
34003400
int expectedNumberOfTensorParameters,
3401-
int expectedNumberOfTensorVariables,
3401+
int expectedNumberOfFunctionTensorVariables,
34023402
int... expectedTensorParameterValueNumbers)
34033403
throws ClassHierarchyException, CancelException, IOException {
34043404
List<File> pathFiles = this.getPathFiles(pythonPath);
@@ -3430,16 +3430,16 @@ private void test(
34303430
LOGGER.info(
34313431
() -> "Pointer key: " + k + " has analysis error: " + v + " at " + v.position()));
34323432

3433-
// Create a mapping from function signatures to pointer keys.
3433+
// a mapping from function signatures to pointer keys.
34343434
Map<String, Set<LocalPointerKey>> functionSignatureToPointerKeys = new HashMap<>();
34353435

3436-
// Create a mapping from function signatures to tensor variables.
3436+
// a mapping from function signatures to tensor variables.
34373437
Map<String, Set<TensorVariable>> functionSignatureToTensorVariables = new HashMap<>();
34383438

34393439
// for each pointer key, tensor variable pair.
34403440
analysis.forEach(
3441-
p -> {
3442-
PointerKey pointerKey = p.fst;
3441+
pt -> {
3442+
PointerKey pointerKey = pt.fst;
34433443

34443444
if (pointerKey instanceof LocalPointerKey) {
34453445
LocalPointerKey localPointerKey = (LocalPointerKey) pointerKey;
@@ -3460,7 +3460,7 @@ private void test(
34603460
return v;
34613461
});
34623462

3463-
TensorVariable tensorVariable = p.snd;
3463+
TensorVariable tensorVariable = pt.snd;
34643464

34653465
// associate the method to the tensor variables.
34663466
functionSignatureToTensorVariables.compute(
@@ -3491,9 +3491,9 @@ private void test(
34913491
Set<TensorVariable> functionTensorVariables =
34923492
functionSignatureToTensorVariables.getOrDefault(functionSignature, emptySet());
34933493

3494-
assertEquals(expectedNumberOfTensorVariables, functionTensorVariables.size());
3494+
assertEquals(expectedNumberOfFunctionTensorVariables, functionTensorVariables.size());
34953495

3496-
// check value numbers.
3496+
// check value number cardinality.
34973497
assertEquals(
34983498
"Each tensor parameter should have a unique value number.",
34993499
expectedNumberOfTensorParameters,
@@ -3517,7 +3517,7 @@ private void test(
35173517

35183518
assertEquals(expectedNumberOfTensorParameters, functionParameterPointerKeys.size());
35193519

3520-
// check value numbers.
3520+
// check actual value numbers.
35213521
Set<Integer> actualParameterValueNumberSet =
35223522
functionParameterPointerKeys.stream()
35233523
.map(LocalPointerKey::getValueNumber)

0 commit comments

Comments
 (0)