From 69ce28e306e184893224e060d9c18594f1be36ff Mon Sep 17 00:00:00 2001 From: Charles Cooper Date: Fri, 13 Dec 2024 10:28:11 -0500 Subject: [PATCH] some more test cleanup --- tests/functional/syntax/test_import.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/tests/functional/syntax/test_import.py b/tests/functional/syntax/test_import.py index 6baa52e8c0..da672c0de5 100644 --- a/tests/functional/syntax/test_import.py +++ b/tests/functional/syntax/test_import.py @@ -27,8 +27,9 @@ def foo(): {"top.vy": top, "subdir0/lib0.vy": lib0, "subdir0/subdir1/lib1.vy": lib1} ) + file_input = input_bundle.load_file("top.vy") with pytest.raises(ModuleNotFound): - compiler.compile_code(top, input_bundle=input_bundle) + compiler.compile_from_file_input(file_input, input_bundle=input_bundle) lib0 = """ from subdir1 import lib1 as lib1 @@ -40,8 +41,9 @@ def foo(): {"top.vy": top, "subdir0/lib0.vy": lib0, "subdir0/subdir1/lib1.vy": lib1} ) + file_input = input_bundle.load_file("top.vy") with pytest.raises(ModuleNotFound): - compiler.compile_code(top, input_bundle=input_bundle) + compiler.compile_from_file_input(file_input, input_bundle=input_bundle) def test_relative_import_searches_only_current_path(make_input_bundle):