update test to use output method

crytic · Feb 6, 2024 · 9f255d8 · 9f255d8
1 parent abe0036
commit 9f255d8
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/tests/e2e/printers/test_printers.py b/tests/e2e/printers/test_printers.py
@@ -31,9 +31,8 @@ def test_inheritance_printer(solc_binary_path) -> None:
     slither = Slither(compilation)
     printer = PrinterInheritanceGraph(slither=slither, logger=None)
 
-    content = ""
-    for contract in slither.contracts:
-        content += printer._summary(contract)
+    output = printer.output('test_printer.dot')
+    content = output.elements[0]['name']['content']
 
     pattern = re.compile(r"(?:c\d+_)?(\w+ -> )(?:c\d+_)(\w+)")
     matches = re.findall(pattern, content)