fix

Author: ciaranra

python/quantum-pecos/tests/guppy/test_multi_module_handling.py

@@ -100,7 +100,7 @@ def single_hadamard() -> bool:
 
     print(f"Single module test - Modules: {module_count}, Functions: {function_names}")
     assert module_count >= 1, "Should have at least one module"
-    assert "single_hadamard" in function_names, "Should contain the main function"
+    assert any(fn.endswith("single_hadamard") for fn in function_names), "Should contain the main function"
 
 
 def test_multiple_functions_compilation() -> None:
@@ -143,8 +143,10 @@ def single_qubit_test() -> bool:
     assert bell_modules >= 1, "Bell pair should have at least one module"
     assert single_modules >= 1, "Single qubit should have at least one module"
 
-    assert "create_bell_pair" in bell_functions, "Bell HUGR should contain create_bell_pair"
-    assert "single_qubit_test" in single_functions, "Single HUGR should contain single_qubit_test"
+    assert any(fn.endswith("create_bell_pair") for fn in bell_functions), "Bell HUGR should contain create_bell_pair"
+    assert any(
+        fn.endswith("single_qubit_test") for fn in single_functions
+    ), "Single HUGR should contain single_qubit_test"
 
 
 def test_compiler_comparison_simple() -> None:

python/quantum-pecos/tests/selene/test_hugr_to_ast.py

@@ -179,7 +179,7 @@ def my_quantum_circuit() -> bool:
             return measure(q)
 
         ast = guppy_to_ast(my_quantum_circuit)
-        assert ast.name == "my_quantum_circuit"
+        assert ast.name.endswith("my_quantum_circuit")
 
     def test_custom_allocator_name(self) -> None:
         """Test custom allocator name."""