From c017371256ae45be1334bb54568efb5eb28d64c0 Mon Sep 17 00:00:00 2001 From: Harry Kalogirou Date: Sat, 9 Mar 2024 23:00:26 +0200 Subject: [PATCH] update test for venom --- tests/unit/compiler/asm/test_asm_optimizer.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/tests/unit/compiler/asm/test_asm_optimizer.py b/tests/unit/compiler/asm/test_asm_optimizer.py index ce32249202..d1edd97189 100644 --- a/tests/unit/compiler/asm/test_asm_optimizer.py +++ b/tests/unit/compiler/asm/test_asm_optimizer.py @@ -95,7 +95,7 @@ def test_dead_code_eliminator(code): assert all(ctor_only not in instr for instr in runtime_asm) -def test_library_code_eliminator(make_input_bundle): +def test_library_code_eliminator(make_input_bundle, venom_pipeline): library = """ @internal def unused1(): @@ -119,6 +119,7 @@ def foo(): input_bundle = make_input_bundle({"library.vy": library}) res = compile_code(code, input_bundle=input_bundle, output_formats=["asm"]) asm = res["asm"] - assert "some_function()" in asm + if not venom_pipeline: + assert "some_function()" in asm assert "unused1()" not in asm assert "unused2()" not in asm