fix lint, some tests

vyperlang · Dec 31, 2023 · 8ac8fe6 · 8ac8fe6
1 parent 3fb3bc0
commit 8ac8fe6
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/tests/functional/codegen/test_call_graph_stability.py b/tests/functional/codegen/test_call_graph_stability.py
@@ -54,7 +54,7 @@ def foo():
         t = CompilerData(code)
 
         # check the .called_functions data structure on foo() directly
-        foo = t.vyper_module_annotated.get_children(vy_ast.FunctionDef, filters={"name": "foo"})[0]
+        foo = t.annotated_vyper_module.get_children(vy_ast.FunctionDef, filters={"name": "foo"})[0]
         foo_t = foo._metadata["func_type"]
         assert [f.name for f in foo_t.called_functions] == func_names
 

diff --git a/tests/unit/ast/test_ast_dict.py b/tests/unit/ast/test_ast_dict.py
@@ -41,8 +41,8 @@ def test_basic_ast():
     code = """
 a: int128
     """
-    dict_out = compiler.compile_code(code, output_formats=["ast_dict"], source_id=0)
-    assert dict_out["ast_dict"]["ast"]["body"][0] == {
+    dict_out = compiler.compile_code(code, output_formats=["annotated_ast_dict"], source_id=0)
+    assert dict_out["annotated_ast_dict"]["ast"]["body"][0] == {
         "annotation": {
             "ast_type": "Name",
             "col_offset": 3,

diff --git a/tests/unit/ast/test_natspec.py b/tests/unit/ast/test_natspec.py
@@ -60,7 +60,7 @@ def doesEat(food: String[30], qty: uint256) -> bool:
 
 
 def parse_natspec(code):
-    vyper_ast = CompilerData(code).vyper_module_annotated
+    vyper_ast = CompilerData(code).annotated_vyper_module
     return vy_ast.parse_natspec(vyper_ast)