add wrapper for verifying CG edges

Author: juliandolby

core/com.ibm.wala.cast.python.test/source/com/ibm/wala/cast/python/test/TestPythonCallGraphShape.java

@@ -2,6 +2,17 @@
 
 import static java.util.Collections.emptyList;
 
+import java.io.File;
+import java.io.IOException;
+import java.net.MalformedURLException;
+import java.net.URI;
+import java.net.URISyntaxException;
+import java.net.URL;
+import java.util.ArrayList;
+import java.util.Collection;
+import java.util.List;
+import java.util.Set;
+
 import com.ibm.wala.cast.python.client.PythonAnalysisEngine;
 import com.ibm.wala.cast.python.types.PythonTypes;
 import com.ibm.wala.cast.types.AstMethodReference;
@@ -18,15 +29,6 @@
 import com.ibm.wala.types.TypeReference;
 import com.ibm.wala.util.CancelException;
 import com.ibm.wala.util.collections.HashSetFactory;
-import java.io.File;
-import java.io.IOException;
-import java.net.MalformedURLException;
-import java.net.URI;
-import java.net.URISyntaxException;
-import java.net.URL;
-import java.util.Collection;
-import java.util.List;
-import java.util.Set;
 
 public abstract class TestPythonCallGraphShape extends TestCallGraphShape {
 
@@ -130,4 +132,12 @@ StringBuffer dump(CallGraph CG) {
     }
     return sb;
   }
+  
+  protected void verifyGraphAssertions(CallGraph CG, Object[][] data) {
+	  List<GraphAssertion> asserts = new ArrayList<>();
+	  for(int i = 0; i < data.length; i++) {
+		  asserts.add(new GraphAssertion(data[i][0], (String[]) data[i][1]));
+	  }
+	  verifyGraphAssertions(CG, asserts);
+  }
 }