Removed clear_registry function

BQSKit · Aug 28, 2024 · 6e31654 · 6e31654
1 parent b6219a9
commit 6e31654
Showing 1 changed file with 13 additions and 6 deletions.
diff --git a/tests/compiler/test_registry.py b/tests/compiler/test_registry.py
@@ -10,7 +10,6 @@
 from bqskit.compiler import compile
 from bqskit.compiler.machine import MachineModel
 from bqskit.compiler.registry import _workflow_registry
-from bqskit.compiler.registry import clear_registry
 from bqskit.compiler.registry import register_workflow
 from bqskit.compiler.workflow import Workflow
 from bqskit.compiler.workflow import WorkflowLike
@@ -74,26 +73,30 @@ class TestRegisterWorkflow:
 
     @pytest.fixture(autouse=True)
     def setup(self) -> None:
-        # _workflow_registry.clear()
-        clear_registry()
+        assert _workflow_registry == _workflow_registry
+        global _workflow_registry
+        _workflow_registry.clear()
 
     def test_register_workflow(self) -> None:
+        global _workflow_registry
         assert _workflow_registry == {}
         gateset = [CZGate(), HGate(), RZGate()]
         num_qudits = 3
         machine = MachineModel(num_qudits, gate_set=gateset)
         workflow = [QuickPartitioner(), ScanningGateRemovalPass()]
-        register_workflow(machine, workflow)
+        register_workflow(machine, workflow, 1)
         assert machine in _workflow_registry
         assert 1 in _workflow_registry[machine]
         assert workflow_match(_workflow_registry[machine][1], workflow)
 
     def test_custom_compile_machine(self) -> None:
+        global _workflow_registry
+        assert _workflow_registry == {}
         gateset = [CZGate(), HGate(), RZGate()]
         num_qudits = 3
         machine = MachineModel(num_qudits, gate_set=gateset)
         workflow = [QuickPartitioner(2)]
-        register_workflow(machine, workflow)
+        register_workflow(machine, workflow, 1)
         circuit = simple_circuit(num_qudits, gateset)
         result = compile(circuit, machine)
         assert unitary_match(result, circuit)
@@ -103,11 +106,13 @@ def test_custom_compile_machine(self) -> None:
         assert result.gate_counts == circuit.gate_counts
 
     def test_custom_compile_gateset(self) -> None:
+        global _workflow_registry
+        assert _workflow_registry == {}
         gateset = [CZGate(), HGate(), RZGate()]
         num_qudits = 3
         machine = MachineModel(num_qudits, gate_set=gateset)
         workflow = [QuickPartitioner(2)]
-        register_workflow(gateset, workflow)
+        register_workflow(gateset, workflow, 1)
         circuit = simple_circuit(num_qudits, gateset)
         result = compile(circuit, machine)
         assert unitary_match(result, circuit)
@@ -117,6 +122,8 @@ def test_custom_compile_gateset(self) -> None:
         assert result.gate_counts == circuit.gate_counts
 
     def test_custom_opt_level(self) -> None:
+        global _workflow_registry
+        assert _workflow_registry == {}
         gateset = [CZGate(), HGate(), RZGate()]
         num_qudits = 3
         machine = MachineModel(num_qudits, gate_set=gateset)