fix pybind tests

quantumlib · Dec 7, 2023 · 73c3d79 · 73c3d79
1 parent 62c0c1a
commit 73c3d79
Showing 1 changed file with 21 additions and 14 deletions.
diff --git a/src/stim/simulators/frame_simulator_pybind_test.py b/src/stim/simulators/frame_simulator_pybind_test.py
@@ -261,44 +261,51 @@ def test_apply_pauli_errors():
         stim.PauliString("+_ZX"),
         stim.PauliString("+ZYZ")
     ]
+
+    # do it again with ints
+    sim = stim.FlipSimulator(
+        batch_size=2,
+        num_qubits=3,
+        disable_stabilizer_randomization=True,
+    )
     sim.apply_pauli_errors(
-        pauli=1,  # X
+        pauli=1,
         mask=np.asarray([
-            [False, False],
+            [True, False],
             [False, False],
             [True, True]]
         ),
     )
     peek = sim.peek_pauli_flips()
     assert peek == [
-        stim.PauliString("+_Z_"),
-        stim.PauliString("+ZYY")
+        stim.PauliString("+X_X"),
+        stim.PauliString("+__X")
     ]
     sim.apply_pauli_errors(
-        pauli=2,  # Y
+        pauli=3,
         mask=np.asarray([
-            [False, True],
-            [False, True],
-            [False, True]]
+            [True, True],
+            [True, False],
+            [False, False]]
         ),
     )
     peek = sim.peek_pauli_flips()
     assert peek == [
-        stim.PauliString("+_Z_"),
-        stim.PauliString("+X__")
+        stim.PauliString("+YZX"),
+        stim.PauliString("+Z_X")
     ]
     sim.apply_pauli_errors(
-        pauli=3,  # Z
+        pauli=2,
         mask=np.asarray([
-            [False, False],
             [True, False],
+            [False, True],
             [False, True]]
         ),
     )
     peek = sim.peek_pauli_flips()
     assert peek == [
-        stim.PauliString("+___"),
-        stim.PauliString("+Y__")
+        stim.PauliString("+_ZX"),
+        stim.PauliString("+ZYZ")
     ]