Formatting

XanaduAI · Sep 5, 2024 · 6be3a82 · 6be3a82
1 parent fe693f3
commit 6be3a82
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/tests/test_lab_dev/test_transformations/test_cft.py b/tests/test_lab_dev/test_transformations/test_cft.py
@@ -46,6 +46,8 @@ def test_wigner_function(self):
         wigner, _, _ = wigner_discretized(dm, vec, vec)
 
         Wigner = (state >> CFT([0]) >> BtoPS([0], s=0)).representation.ansatz
-        X, Y = np.meshgrid(vec*np.sqrt(2), vec*np.sqrt(2)) #scaling to take care of HBAR
+        X, Y = np.meshgrid(vec * np.sqrt(2), vec * np.sqrt(2))  # scaling to take care of HBAR
         Z = np.array([X - 1j * Y, X + 1j * Y]).transpose((1, 2, 0))
-        assert math.allclose(2*(np.real(Wigner(Z))), (np.real(wigner)), atol=1e-8) #scaling to take care of HBAR
+        assert math.allclose(
+            2 * (np.real(Wigner(Z))), (np.real(wigner)), atol=1e-8
+        )  # scaling to take care of HBAR