Skip to content

Commit 8f69a9e

Browse files
authored
Add call graph logging to test (#124)
Minor logging enhancement.
1 parent 1fd3e0e commit 8f69a9e

File tree

1 file changed

+11
-4
lines changed

1 file changed

+11
-4
lines changed

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

Lines changed: 11 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@
44
import static org.junit.Assert.assertNotNull;
55
import static org.junit.Assert.assertTrue;
66

7+
import com.ibm.wala.cast.ipa.callgraph.CAstCallGraphUtil;
78
import com.ibm.wala.cast.python.client.PythonAnalysisEngine;
89
import com.ibm.wala.cast.python.ipa.callgraph.PythonSSAPropagationCallGraphBuilder;
910
import com.ibm.wala.cast.python.ml.analysis.TensorTypeAnalysis;
@@ -13,6 +14,7 @@
1314
import com.ibm.wala.ipa.callgraph.CallGraph;
1415
import com.ibm.wala.ipa.callgraph.propagation.LocalPointerKey;
1516
import com.ibm.wala.ipa.callgraph.propagation.PointerKey;
17+
import com.ibm.wala.ipa.callgraph.propagation.SSAPropagationCallGraphBuilder;
1618
import com.ibm.wala.ipa.cha.ClassHierarchyException;
1719
import com.ibm.wala.util.CancelException;
1820
import java.io.IOException;
@@ -24,6 +26,7 @@
2426
import java.util.Iterator;
2527
import java.util.Map;
2628
import java.util.Set;
29+
import java.util.logging.Level;
2730
import java.util.logging.Logger;
2831
import java.util.stream.Collectors;
2932
import org.junit.Test;
@@ -226,10 +229,14 @@ private void testTf2(
226229
CallGraph CG = builder.makeCallGraph(builder.getOptions());
227230
assertNotNull(CG);
228231

229-
// CAstCallGraphUtil.AVOID_DUMP = false;
230-
// CAstCallGraphUtil.dumpCG(((SSAPropagationCallGraphBuilder)builder).getCFAContextInterpreter(),
231-
// builder.getPointerAnalysis(), CG);
232-
// System.err.println(CG);
232+
if (logger.isLoggable(Level.FINE)) {
233+
CAstCallGraphUtil.AVOID_DUMP = false;
234+
CAstCallGraphUtil.dumpCG(
235+
((SSAPropagationCallGraphBuilder) builder).getCFAContextInterpreter(),
236+
builder.getPointerAnalysis(),
237+
CG);
238+
logger.fine("Call graph:\n" + CG);
239+
}
233240

234241
TensorTypeAnalysis analysis = E.performAnalysis(builder);
235242

0 commit comments

Comments
 (0)