diff --git a/tests/passes/partitioning/test_measurement.py b/tests/passes/partitioning/test_measurement.py index b957d5d18..d0faab180 100644 --- a/tests/passes/partitioning/test_measurement.py +++ b/tests/passes/partitioning/test_measurement.py @@ -2,8 +2,8 @@ from bqskit.compiler.compiler import Compiler from bqskit.ir.circuit import Circuit -from bqskit.ir.gates import MeasurementPlaceholder from bqskit.ir.gates import CXGate +from bqskit.ir.gates import MeasurementPlaceholder from bqskit.passes.partitioning.quick import QuickPartitioner @@ -13,10 +13,13 @@ def test_barrier_stop_partitioning_across_some_circuit( circuit = Circuit(4) circuit.append_gate(CXGate(), (0, 1)) measurements = {0: ('c', 0), 1: ('c', 1)} - circuit.append_gate(MeasurementPlaceholder([('c', 2)], measurements), (0, 1)) + circuit.append_gate( + MeasurementPlaceholder( + [('c', 2)], measurements, + ), (0, 1), + ) circuit.append_gate(CXGate(), (0, 1)) circuit.append_gate(CXGate(), (2, 3)) circuit.append_gate(CXGate(), (2, 3)) circuit = compiler.compile(circuit, [QuickPartitioner(2)]) assert circuit.num_operations == 4 - diff --git a/tests/passes/partitioning/test_reset.py b/tests/passes/partitioning/test_reset.py index 9dd0abc3e..2a8dadada 100644 --- a/tests/passes/partitioning/test_reset.py +++ b/tests/passes/partitioning/test_reset.py @@ -2,8 +2,8 @@ from bqskit.compiler.compiler import Compiler from bqskit.ir.circuit import Circuit -from bqskit.ir.gates import Reset from bqskit.ir.gates import CXGate +from bqskit.ir.gates import Reset from bqskit.passes.partitioning.quick import QuickPartitioner @@ -18,4 +18,3 @@ def test_reset_stop_partitioning_across_some_circuit( circuit.append_gate(CXGate(), (2, 3)) circuit = compiler.compile(circuit, [QuickPartitioner(2)]) assert circuit.num_operations == 4 -