Fix test_linear_synthesis

Tarun-Kumar07 · Apr 17, 2024 · b254b3b · b254b3b
1 parent 0e81d05
commit b254b3b
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/test/python/synthesis/test_linear_synthesis.py b/test/python/synthesis/test_linear_synthesis.py
@@ -94,11 +94,19 @@ def test_example_circuit(self):
         qc.cx(1, 0)
         mat = LinearFunction(qc).linear
 
-        optimized_qc = optimize_cx_4_options(synth_cnot_count_full_pmh, mat, optimize_count=True)
+        synth_cnot_pmh_section_size_2 = lambda state: synth_cnot_count_full_pmh(
+            state, section_size=2
+        )
+
+        optimized_qc = optimize_cx_4_options(
+            synth_cnot_pmh_section_size_2, mat, optimize_count=True
+        )
         self.assertEqual(optimized_qc.depth(), 17)
         self.assertEqual(optimized_qc.count_ops()["cx"], 20)
 
-        optimized_qc = optimize_cx_4_options(synth_cnot_count_full_pmh, mat, optimize_count=False)
+        optimized_qc = optimize_cx_4_options(
+            synth_cnot_pmh_section_size_2, mat, optimize_count=False
+        )
         self.assertEqual(optimized_qc.depth(), 15)
         self.assertEqual(optimized_qc.count_ops()["cx"], 23)