bugfix in CZ test

XanaduAI · Nov 7, 2024 · ed7a7bd · ed7a7bd
1 parent bdecc2f
commit ed7a7bd
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/tests/test_lab_dev/test_transformations/test_czgate.py b/tests/test_lab_dev/test_transformations/test_czgate.py
@@ -48,8 +48,7 @@ def test_init(self):
     @pytest.mark.parametrize("s", [0.1, 0.2, 1.5])
     def test_application(self, s):
         "Tests the application of CZgate"
-        psi = Coherent([0], 0, 1) >> Coherent([1], 1, 0) >> CZgate([0, 1], 1)
+        psi = Coherent([0], 0, 1) >> Coherent([1], 1, 0) >> CZgate([0, 1], s)
         _, d, _ = psi.phase_space(s=0)
-        psi = Coherent([0], 0, 1) >> Coherent([1], 1, 0) >> CZgate([0, 1], 1)
         d_by_hand = math.astensor([0, math.sqrt(complex(2)), (1 + s) * math.sqrt(complex(2)), 0])
         assert math.allclose(d[0], d_by_hand)