def test_replace_subscripts_nested(): test_ast = vy_ast.parse_to_ast("[[0, 1], [2, 3], [4, 5]][2][1]") expected_ast = vy_ast.parse_to_ast("5") folding.replace_subscripts(test_ast) assert vy_ast.compare_nodes(test_ast, expected_ast)
def test_replace_subscripts_simple(): test_ast = vy_ast.parse_to_ast("[foo, bar, baz][1]") expected_ast = vy_ast.parse_to_ast("bar") folding.replace_subscripts(test_ast) assert vy_ast.compare_nodes(test_ast, expected_ast)