Skip to content

Add file-based testing infrastructure #4761

New issue
New issue
Open
Open
Add file-based testing infrastructure#4761
@Sporarum

Description

@Sporarum
Sporarum
opened on Oct 30, 2025

Current problem

Tests like tests/unit/ast/test_ast_dict.py::test_output_type_info are complicated to understand and update

vyper/tests/unit/ast/test_ast_dict.py

Lines 173 to 1283 in db8abe8

def test_output_type_info(make_input_bundle, chdir_tmp_path):
# test type info is output in the ast dict
# test different, complex types, and test import info is also output
lib1 = """
# add a pragma to see if it gets added properly to the module's settings object
#pragma enable-decimals
struct Foo:
x: uint256
event Bar:
pass
struct Baz:
x: decimal
y: Bytes[20]
z: String[32]
w: uint256
u: address
interface Qux:
def return_tuple() -> (Foo[1], uint256): nonpayable
foo_var: Foo
sarray_var: Foo[1]
darray_var: DynArray[Foo, 5]
interface_var: Qux
hashmap_var: HashMap[address, Foo]
sarray_var2: uint256[2]
darray_var2: DynArray[uint256, 5]
@internal
def foo():
t: uint256 = max_value(uint256)
u: int24 = empty(int24)
self.foo_var = empty(Foo)
self.sarray_var[0] = empty(Foo)
self.darray_var[1] = empty(Foo)
self.sarray_var, t = extcall self.interface_var.return_tuple()
@external
def bar():
s: bytes24 = empty(bytes24)
"""
main = """
import lib1
initializes: lib1
@internal
def foo():
lib1.foo()
log lib1.Bar()
s: lib1.Foo = empty(lib1.Foo)
"""
input_bundle = make_input_bundle({"lib1.vy": lib1, "main.vy": main})
lib1_file = input_bundle.load_file("lib1.vy")
lib1_out = compiler.compile_from_file_input(
lib1_file, input_bundle=input_bundle, output_formats=["annotated_ast_dict"]
)
lib1_ast = copy.deepcopy(lib1_out["annotated_ast_dict"]["ast"])
lib1_sha256sum = lib1_ast.pop("source_sha256sum")
assert lib1_sha256sum == lib1_file.sha256sum
to_strip = NODE_SRC_ATTRIBUTES + ("resolved_path", "variable_reads", "variable_writes")
_strip_source_annotations(lib1_ast, to_strip=to_strip)
main_file = input_bundle.load_file("main.vy")
main_out = compiler.compile_from_file_input(
main_file, input_bundle=input_bundle, output_formats=["annotated_ast_dict"]
)
main_ast = main_out["annotated_ast_dict"]["ast"]
main_sha256sum = main_ast.pop("source_sha256sum")
assert main_sha256sum == main_file.sha256sum
_strip_source_annotations(main_ast, to_strip=to_strip)
assert main_out["annotated_ast_dict"]["imports"][0] == lib1_out["annotated_ast_dict"]["ast"]
# TODO: would be nice to refactor this into bunch of small test cases
assert main_ast == {
"ast_type": "Module",
"settings": {},
"body": [
{
"alias": None,
"ast_type": "Import",
"import_info": {
"alias": "lib1",
"file_sha256sum": lib1_file.sha256sum,
"path": "lib1.vy",
"qualified_module_name": "lib1",
"source_id": 0,
},
"name": "lib1",
"node_id": 1,
},
{
"annotation": {"ast_type": "Name", "id": "lib1", "node_id": 6},
"ast_type": "InitializesDecl",
"node_id": 3,
},
{
"args": {
"args": [],
"ast_type": "arguments",
"default": None,
"defaults": [],
"node_id": 9,
},
"ast_type": "FunctionDef",
"body": [
{
"ast_type": "Expr",
"node_id": 10,
"value": {
"args": [],
"ast_type": "Call",
"func": {
"ast_type": "Attribute",
"attr": "foo",
"node_id": 12,
"type": {
"name": "foo",
"type_decl_node": {"node_id": 119, "source_id": 0},
"typeclass": "contract_function",
},
"value": {
"ast_type": "Name",
"id": "lib1",
"node_id": 13,
"type": {
"name": "lib1.vy",
"type_decl_node": {"node_id": 0, "source_id": 0},
"typeclass": "module",
},
},
},
"keywords": [],
"node_id": 11,
"type": {"name": "(void)"},
},
},
{
"ast_type": "Log",
"node_id": 17,
"type": {
"name": "Bar",
"type_decl_node": {"node_id": 7, "source_id": 0},
"typeclass": "event",
},
"value": {
"args": [],
"ast_type": "Call",
"func": {
"ast_type": "Attribute",
"attr": "Bar",
"node_id": 19,
"type": {
"type_t": {
"name": "Bar",
"type_decl_node": {"node_id": 7, "source_id": 0},
"typeclass": "event",
}
},
"value": {
"ast_type": "Name",
"id": "lib1",
"node_id": 20,
"type": {
"name": "lib1.vy",
"type_decl_node": {"node_id": 0, "source_id": 0},
"typeclass": "module",
},
},
},
"keywords": [],
"node_id": 18,
"type": {"name": "(void)"},
},
},
{
"annotation": {
"ast_type": "Attribute",
"attr": "Foo",
"node_id": 26,
"value": {"ast_type": "Name", "id": "lib1", "node_id": 27},
},
"ast_type": "AnnAssign",
"node_id": 23,
"target": {
"ast_type": "Name",
"id": "s",
"node_id": 24,
"type": {"name": "Foo", "typeclass": "struct"},
},
"value": {
"args": [
{
"ast_type": "Attribute",
"attr": "Foo",
"node_id": 33,
"type": {"type_t": {"name": "Foo", "typeclass": "struct"}},
"value": {
"ast_type": "Name",
"id": "lib1",
"node_id": 34,
"type": {
"name": "lib1.vy",
"type_decl_node": {"node_id": 0, "source_id": 0},
"typeclass": "module",
},
},
}
],
"ast_type": "Call",
"func": {
"ast_type": "Name",
"id": "empty",
"node_id": 31,
"type": {"name": "empty", "typeclass": "builtin_function"},
},
"keywords": [],
"node_id": 30,
"type": {"name": "Foo", "typeclass": "struct"},
},
},
],
"decorator_list": [{"ast_type": "Name", "id": "internal", "node_id": 37}],
"doc_string": None,
"name": "foo",
"node_id": 8,
"pos": None,
"returns": None,
},
],
"doc_string": None,
"name": None,
"node_id": 0,
"path": "main.vy",
"source_id": 1,
"is_interface": False,
"type": {
"name": "main.vy",
"type_decl_node": {"node_id": 0, "source_id": 1},
"typeclass": "module",
},
}
# TODO: would be nice to refactor this into bunch of small test cases
# TODO: write the test in a way which makes the links between nodes
# clearer
assert lib1_ast == {
"ast_type": "Module",
"body": [
{
"ast_type": "StructDef",
"body": [
{
"annotation": {"ast_type": "Name", "id": "uint256", "node_id": 5},
"ast_type": "AnnAssign",
"node_id": 2,
"target": {"ast_type": "Name", "id": "x", "node_id": 3},
"value": None,
}
],
"doc_string": None,
"name": "Foo",
"node_id": 1,
},
{
"ast_type": "EventDef",
"body": [{"ast_type": "Pass", "node_id": 8}],
"doc_string": None,
"name": "Bar",
"node_id": 7,
},
{
"ast_type": "StructDef",
"body": [
{
"annotation": {"ast_type": "Name", "id": "decimal", "node_id": 13},
"ast_type": "AnnAssign",
"node_id": 10,
"target": {"ast_type": "Name", "id": "x", "node_id": 11},
"value": None,
},
{
"annotation": {
"ast_type": "Subscript",
"node_id": 18,
"slice": {
"ast_type": "Int",
"node_id": 21,
"value": 20,
"folded_value": {"ast_type": "Int", "node_id": 21, "value": 20},
},
"value": {"ast_type": "Name", "id": "Bytes", "node_id": 19},
},
"ast_type": "AnnAssign",
"node_id": 15,
"target": {"ast_type": "Name", "id": "y", "node_id": 16},
"value": None,
},
{
"annotation": {
"ast_type": "Subscript",
"node_id": 26,
"slice": {
"ast_type": "Int",
"node_id": 29,
"value": 32,
"folded_value": {"ast_type": "Int", "node_id": 29, "value": 32},
},
"value": {"ast_type": "Name", "id": "String", "node_id": 27},
},
"ast_type": "AnnAssign",
"node_id": 23,
"target": {"ast_type": "Name", "id": "z", "node_id": 24},
"value": None,
},
{
"annotation": {"ast_type": "Name", "id": "uint256", "node_id": 34},
"ast_type": "AnnAssign",
"node_id": 31,
"target": {"ast_type": "Name", "id": "w", "node_id": 32},
"value": None,
},
{
"annotation": {"ast_type": "Name", "id": "address", "node_id": 39},
"ast_type": "AnnAssign",
"node_id": 36,
"target": {"ast_type": "Name", "id": "u", "node_id": 37},
"value": None,
},
],
"doc_string": None,
"name": "Baz",
"node_id": 9,
},
{
"ast_type": "InterfaceDef",
"body": [
{
"args": {
"args": [],
"ast_type": "arguments",
"default": None,
"defaults": [],
"node_id": 43,
},
"ast_type": "FunctionDef",
"body": [
{
"ast_type": "Expr",
"node_id": 44,
"value": {"ast_type": "Name", "id": "nonpayable", "node_id": 45},
}
],
"decorator_list": [],
"doc_string": None,
"name": "return_tuple",
"node_id": 42,
"pos": None,
"returns": {
"ast_type": "Tuple",
"elements": [
{
"ast_type": "Subscript",
"node_id": 48,
"slice": {
"ast_type": "Int",
"node_id": 51,
"value": 1,
"folded_value": {
"ast_type": "Int",
"node_id": 51,
"value": 1,
},
},
"value": {"ast_type": "Name", "id": "Foo", "node_id": 49},
},
{"ast_type": "Name", "id": "uint256", "node_id": 53},
],
"node_id": 47,
},
}
],
"doc_string": None,
"name": "Qux",
"node_id": 41,
},
{
"annotation": {"ast_type": "Name", "id": "Foo", "node_id": 59},
"ast_type": "VariableDecl",
"is_constant": False,
"is_immutable": False,
"is_public": False,
"is_reentrant": False,
"is_transient": False,
"node_id": 56,
"target": {
"ast_type": "Name",
"id": "foo_var",
"node_id": 57,
"type": {"name": "Foo", "typeclass": "struct"},
},
"value": None,
"type": {"name": "Foo", "typeclass": "struct"},
},
{
"annotation": {
"ast_type": "Subscript",
"node_id": 64,
"slice": {
"ast_type": "Int",
"node_id": 67,
"value": 1,
"folded_value": {"ast_type": "Int", "node_id": 67, "value": 1},
},
"value": {"ast_type": "Name", "id": "Foo", "node_id": 65},
},
"ast_type": "VariableDecl",
"is_constant": False,
"is_immutable": False,
"is_public": False,
"is_reentrant": False,
"is_transient": False,
"node_id": 61,
"target": {
"ast_type": "Name",
"id": "sarray_var",
"node_id": 62,
"type": {
"length": 1,
"value_type": {"name": "Foo", "typeclass": "struct"},
"name": "$SArray",
"typeclass": "static_array",
},
},
"value": None,
"type": {
"length": 1,
"value_type": {"name": "Foo", "typeclass": "struct"},
"name": "$SArray",
"typeclass": "static_array",
},
},
{
"annotation": {
"ast_type": "Subscript",
"node_id": 72,
"slice": {
"ast_type": "Tuple",
"elements": [
{"ast_type": "Name", "id": "Foo", "node_id": 76},
{
"ast_type": "Int",
"node_id": 78,
"value": 5,
"folded_value": {"ast_type": "Int", "node_id": 78, "value": 5},
},
],
"node_id": 75,
},
"value": {"ast_type": "Name", "id": "DynArray", "node_id": 73},
},
"ast_type": "VariableDecl",
"is_constant": False,
"is_immutable": False,
"is_public": False,
"is_reentrant": False,
"is_transient": False,
"node_id": 69,
"target": {
"ast_type": "Name",
"id": "darray_var",
"node_id": 70,
"type": {
"length": 5,
"value_type": {"name": "Foo", "typeclass": "struct"},
"name": "DynArray",
"typeclass": "dynamic_array",
},
},
"value": None,
"type": {
"length": 5,
"value_type": {"name": "Foo", "typeclass": "struct"},
"name": "DynArray",
"typeclass": "dynamic_array",
},
},
{
"annotation": {"ast_type": "Name", "id": "Qux", "node_id": 84},
"ast_type": "VariableDecl",
"is_constant": False,
"is_immutable": False,
"is_public": False,
"is_reentrant": False,
"is_transient": False,
"node_id": 81,
"target": {
"ast_type": "Name",
"id": "interface_var",
"node_id": 82,
"type": {
"name": "Qux",
"type_decl_node": {"node_id": 41, "source_id": 0},
"typeclass": "interface",
},
},
"value": None,
"type": {
"name": "Qux",
"type_decl_node": {"node_id": 41, "source_id": 0},
"typeclass": "interface",
},
},
{
"annotation": {
"ast_type": "Subscript",
"node_id": 89,
"slice": {
"ast_type": "Tuple",
"elements": [
{"ast_type": "Name", "id": "address", "node_id": 93},
{"ast_type": "Name", "id": "Foo", "node_id": 95},
],
"node_id": 92,
},
"value": {"ast_type": "Name", "id": "HashMap", "node_id": 90},
},
"ast_type": "VariableDecl",
"is_constant": False,
"is_immutable": False,
"is_public": False,
"is_reentrant": False,
"is_transient": False,
"node_id": 86,
"target": {
"ast_type": "Name",
"id": "hashmap_var",
"node_id": 87,
"type": {
"key_type": {"name": "address"},
"value_type": {"name": "Foo", "typeclass": "struct"},
"name": "HashMap",
"typeclass": "hashmap",
},
},
"value": None,
"type": {
"key_type": {"name": "address"},
"value_type": {"name": "Foo", "typeclass": "struct"},
"name": "HashMap",
"typeclass": "hashmap",
},
},
{
"annotation": {
"ast_type": "Subscript",
"node_id": 102,
"slice": {
"ast_type": "Int",
"node_id": 105,
"value": 2,
"folded_value": {"ast_type": "Int", "node_id": 105, "value": 2},
},
"value": {"ast_type": "Name", "id": "uint256", "node_id": 103},
},
"ast_type": "VariableDecl",
"is_constant": False,
"is_immutable": False,
"is_public": False,
"is_reentrant": False,
"is_transient": False,
"node_id": 99,
"target": {
"ast_type": "Name",
"id": "sarray_var2",
"node_id": 100,
"type": {
"length": 2,
"value_type": {
"bits": 256,
"is_signed": False,
"name": "uint256",
"typeclass": "integer",
},
"name": "$SArray",
"typeclass": "static_array",
},
},
"value": None,
"type": {
"length": 2,
"value_type": {
"bits": 256,
"is_signed": False,
"name": "uint256",
"typeclass": "integer",
},
"name": "$SArray",
"typeclass": "static_array",
},
},
{
"annotation": {
"ast_type": "Subscript",
"node_id": 110,
"slice": {
"ast_type": "Tuple",
"elements": [
{"ast_type": "Name", "id": "uint256", "node_id": 114},
{
"ast_type": "Int",
"node_id": 116,
"value": 5,
"folded_value": {"ast_type": "Int", "node_id": 116, "value": 5},
},
],
"node_id": 113,
},
"value": {"ast_type": "Name", "id": "DynArray", "node_id": 111},
},
"ast_type": "VariableDecl",
"is_constant": False,
"is_immutable": False,
"is_public": False,
"is_reentrant": False,
"is_transient": False,
"node_id": 107,
"target": {
"ast_type": "Name",
"id": "darray_var2",
"node_id": 108,
"type": {
"length": 5,
"value_type": {
"bits": 256,
"is_signed": False,
"name": "uint256",
"typeclass": "integer",
},
"name": "DynArray",
"typeclass": "dynamic_array",
},
},
"value": None,
"type": {
"length": 5,
"value_type": {
"bits": 256,
"is_signed": False,
"name": "uint256",
"typeclass": "integer",
},
"name": "DynArray",
"typeclass": "dynamic_array",
},
},
{
"args": {
"args": [],
"ast_type": "arguments",
"default": None,
"defaults": [],
"node_id": 120,
},
"ast_type": "FunctionDef",
"body": [
{
"annotation": {"ast_type": "Name", "id": "uint256", "node_id": 124},
"ast_type": "AnnAssign",
"node_id": 121,
"target": {
"ast_type": "Name",
"id": "t",
"node_id": 122,
"type": {
"bits": 256,
"is_signed": False,
"name": "uint256",
"typeclass": "integer",
},
},
"value": {
"args": [
{
"ast_type": "Name",
"id": "uint256",
"node_id": 129,
"type": {
"type_t": {
"bits": 256,
"is_signed": False,
"name": "uint256",
"typeclass": "integer",
}
},
}
],
"ast_type": "Call",
"func": {
"ast_type": "Name",
"id": "max_value",
"node_id": 127,
"type": {"name": "max_value", "typeclass": "builtin_function"},
},
"keywords": [],
"node_id": 126,
"type": {
"bits": 256,
"is_signed": False,
"name": "uint256",
"typeclass": "integer",
},
"folded_value": {
"ast_type": "Int",
"node_id": 126,
"value": 115792089237316195423570985008687907853269984665640564039457584007913129639935, # noqa: E501
"type": {
"bits": 256,
"is_signed": False,
"name": "uint256",
"typeclass": "integer",
},
},
},
},
{
"annotation": {"ast_type": "Name", "id": "int24", "node_id": 134},
"ast_type": "AnnAssign",
"node_id": 131,
"target": {
"ast_type": "Name",
"id": "u",
"node_id": 132,
"type": {
"bits": 24,
"is_signed": True,
"name": "int24",
"typeclass": "integer",
},
},
"value": {
"args": [
{
"ast_type": "Name",
"id": "int24",
"node_id": 139,
"type": {
"type_t": {
"bits": 24,
"is_signed": True,
"name": "int24",
"typeclass": "integer",
}
},
}
],
"ast_type": "Call",
"func": {
"ast_type": "Name",
"id": "empty",
"node_id": 137,
"type": {"name": "empty", "typeclass": "builtin_function"},
},
"keywords": [],
"node_id": 136,
"type": {
"bits": 24,
"is_signed": True,
"name": "int24",
"typeclass": "integer",
},
},
},
{
"ast_type": "Assign",
"node_id": 141,
"target": {
"ast_type": "Attribute",
"attr": "foo_var",
"node_id": 142,
"value": {
"ast_type": "Name",
"id": "self",
"node_id": 143,
"type": {"name": "self"},
},
"type": {"name": "Foo", "typeclass": "struct"},
},
"value": {
"args": [
{
"ast_type": "Name",
"id": "Foo",
"node_id": 149,
"type": {"type_t": {"name": "Foo", "typeclass": "struct"}},
}
],
"ast_type": "Call",
"func": {
"ast_type": "Name",
"id": "empty",
"node_id": 147,
"type": {"name": "empty", "typeclass": "builtin_function"},
},
"keywords": [],
"node_id": 146,
"type": {"name": "Foo", "typeclass": "struct"},
},
},
{
"ast_type": "Assign",
"node_id": 151,
"target": {
"ast_type": "Subscript",
"node_id": 152,
"slice": {
"ast_type": "Int",
"node_id": 157,
"value": 0,
"type": {
"bits": 8,
"is_signed": True,
"name": "int8",
"typeclass": "integer",
},
"folded_value": {
"ast_type": "Int",
"node_id": 157,
"value": 0,
"type": {
"bits": 8,
"is_signed": True,
"name": "int8",
"typeclass": "integer",
},
},
},
"value": {
"ast_type": "Attribute",
"attr": "sarray_var",
"node_id": 153,
"value": {
"ast_type": "Name",
"id": "self",
"node_id": 154,
"type": {"name": "self"},
},
"type": {
"length": 1,
"value_type": {"name": "Foo", "typeclass": "struct"},
"name": "$SArray",
"typeclass": "static_array",
},
},
"type": {"name": "Foo", "typeclass": "struct"},
},
"value": {
"args": [
{
"ast_type": "Name",
"id": "Foo",
"node_id": 162,
"type": {"type_t": {"name": "Foo", "typeclass": "struct"}},
}
],
"ast_type": "Call",
"func": {
"ast_type": "Name",
"id": "empty",
"node_id": 160,
"type": {"name": "empty", "typeclass": "builtin_function"},
},
"keywords": [],
"node_id": 159,
"type": {"name": "Foo", "typeclass": "struct"},
},
},
{
"ast_type": "Assign",
"node_id": 164,
"target": {
"ast_type": "Subscript",
"node_id": 165,
"slice": {
"ast_type": "Int",
"node_id": 170,
"value": 1,
"type": {
"bits": 8,
"is_signed": True,
"name": "int8",
"typeclass": "integer",
},
"folded_value": {
"ast_type": "Int",
"node_id": 170,
"value": 1,
"type": {
"bits": 8,
"is_signed": True,
"name": "int8",
"typeclass": "integer",
},
},
},
"value": {
"ast_type": "Attribute",
"attr": "darray_var",
"node_id": 166,
"value": {
"ast_type": "Name",
"id": "self",
"node_id": 167,
"type": {"name": "self"},
},
"type": {
"length": 5,
"value_type": {"name": "Foo", "typeclass": "struct"},
"name": "DynArray",
"typeclass": "dynamic_array",
},
},
"type": {"name": "Foo", "typeclass": "struct"},
},
"value": {
"args": [
{
"ast_type": "Name",
"id": "Foo",
"node_id": 175,
"type": {"type_t": {"name": "Foo", "typeclass": "struct"}},
}
],
"ast_type": "Call",
"func": {
"ast_type": "Name",
"id": "empty",
"node_id": 173,
"type": {"name": "empty", "typeclass": "builtin_function"},
},
"keywords": [],
"node_id": 172,
"type": {"name": "Foo", "typeclass": "struct"},
},
},
{
"ast_type": "Assign",
"node_id": 177,
"target": {
"ast_type": "Tuple",
"elements": [
{
"ast_type": "Attribute",
"attr": "sarray_var",
"node_id": 179,
"value": {
"ast_type": "Name",
"id": "self",
"node_id": 180,
"type": {"name": "self"},
},
"type": {
"length": 1,
"value_type": {"name": "Foo", "typeclass": "struct"},
"name": "$SArray",
"typeclass": "static_array",
},
},
{
"ast_type": "Name",
"id": "t",
"node_id": 183,
"type": {
"bits": 256,
"is_signed": False,
"name": "uint256",
"typeclass": "integer",
},
},
],
"node_id": 178,
"type": {"members": {}, "name": "$Tuple", "typeclass": "tuple"},
},
"value": {
"ast_type": "ExtCall",
"node_id": 186,
"value": {
"args": [],
"ast_type": "Call",
"func": {
"ast_type": "Attribute",
"attr": "return_tuple",
"node_id": 188,
"value": {
"ast_type": "Attribute",
"attr": "interface_var",
"node_id": 189,
"value": {
"ast_type": "Name",
"id": "self",
"node_id": 190,
"type": {"name": "self"},
},
"type": {
"name": "Qux",
"type_decl_node": {"node_id": 41, "source_id": 0},
"typeclass": "interface",
},
},
"type": {
"name": "return_tuple",
"type_decl_node": {"node_id": 42, "source_id": 0},
"typeclass": "contract_function",
},
},
"keywords": [],
"node_id": 187,
"type": {"members": {}, "name": "$Tuple", "typeclass": "tuple"},
},
"type": {"members": {}, "name": "$Tuple", "typeclass": "tuple"},
},
},
],
"decorator_list": [{"ast_type": "Name", "id": "internal", "node_id": 194}],
"doc_string": None,
"name": "foo",
"node_id": 119,
"pos": None,
"returns": None,
},
{
"args": {
"args": [],
"ast_type": "arguments",
"default": None,
"defaults": [],
"node_id": 197,
},
"ast_type": "FunctionDef",
"body": [
{
"annotation": {"ast_type": "Name", "id": "bytes24", "node_id": 201},
"ast_type": "AnnAssign",
"node_id": 198,
"target": {
"ast_type": "Name",
"id": "s",
"node_id": 199,
"type": {"m": 24, "name": "bytes24", "typeclass": "bytes_m"},
},
"value": {
"args": [
{
"ast_type": "Name",
"id": "bytes24",
"node_id": 206,
"type": {
"type_t": {
"m": 24,
"name": "bytes24",
"typeclass": "bytes_m",
}
},
}
],
"ast_type": "Call",
"func": {
"ast_type": "Name",
"id": "empty",
"node_id": 204,
"type": {"name": "empty", "typeclass": "builtin_function"},
},
"keywords": [],
"node_id": 203,
"type": {"m": 24, "name": "bytes24", "typeclass": "bytes_m"},
},
}
],
"decorator_list": [{"ast_type": "Name", "id": "external", "node_id": 208}],
"doc_string": None,
"name": "bar",
"node_id": 196,
"pos": None,
"returns": None,
},
],
"doc_string": None,
"is_interface": False,
"name": None,
"node_id": 0,
"path": "lib1.vy",
"settings": {"enable_decimals": True},
"source_id": 0,
"type": {
"name": "lib1.vy",
"type_decl_node": {"node_id": 0, "source_id": 0},
"typeclass": "module",
},
}

Proposed solution

Add some testing infrastructure which puts both test inputs and outputs into files, to make it much easier to add and update tests of this type

Details

Here is a draft of the structure:

tests
  ...
  file-based
    ast
      test_ast_file.py # contains the pytest logic
      foo
        lib1.py # input files
        main.py
        lib1.out # expected ast output files
        main.out
        lib1.out.tmp # actual ast output files, untracked
        main.out.tmp

When the test is run, it first generates .out.tmp files and then checks them against the .out files
This allows us to easily update the test by something like cp *.out.tmp *.out

Optional

Also add a -f option which outputs the same ast output file(s), or a simplified view of them (for example without all the position information)

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions