black on tests

PennyLaneAI · May 2, 2024 · abcff97 · abcff97
1 parent 5384244
commit abcff97
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tests/new_api/test_device.py b/tests/new_api/test_device.py
@@ -456,7 +456,9 @@ def test_custom_wires(self, phi, theta, wires):
         assert np.allclose(result[0], np.cos(phi))
         assert np.allclose(result[1], np.cos(phi) * np.cos(theta))
 
-    @pytest.mark.parametrize("wires, wire_order", [(3, (0,1,2)), (("a", "b", "c"), ("a", "b", "c"))])
+    @pytest.mark.parametrize(
+        "wires, wire_order", [(3, (0, 1, 2)), (("a", "b", "c"), ("a", "b", "c"))]
+    )
     def test_probs_different_wire_orders(self, wires, wire_order):
         """Test that measuring probabilities works with custom wires."""
 
@@ -477,6 +479,7 @@ def test_probs_different_wire_orders(self, wires, wire_order):
         res3 = dev.execute(tape3)
         assert qml.math.allclose(res3, np.array([0.5, 0.0, 0.5, 0.0]))
 
+
 @pytest.mark.parametrize("batch_obs", [True, False])
 class TestDerivatives:
     """Unit tests for calculating derivatives with a device"""