Fix printing difference in doctest

quantumlib · Jun 24, 2024 · 0cd5894 · 0cd5894
1 parent 5d54a81
commit 0cd5894
Showing 1 changed file with 10 additions and 6 deletions.
diff --git a/src/stim/simulators/tableau_simulator.pybind.cc b/src/stim/simulators/tableau_simulator.pybind.cc
@@ -319,15 +319,19 @@ void stim_pybind::pybind_tableau_simulator_methods(
                 >>> import numpy as np
                 >>> s = stim.TableauSimulator()
                 >>> s.x(2)
-                >>> list(s.state_vector(endian='little'))
-                [0j, 0j, 0j, 0j, (1+0j), 0j, 0j, 0j]
+                >>> s.state_vector(endian='little')
+                array([0.+0.j, 0.+0.j, 0.+0.j, 0.+0.j, 1.+0.j, 0.+0.j, 0.+0.j, 0.+0.j],
+                      dtype=complex64)
 
-                >>> list(s.state_vector(endian='big'))
-                [0j, (1+0j), 0j, 0j, 0j, 0j, 0j, 0j]
+
+                >>> s.state_vector(endian='big')
+                array([0.+0.j, 1.+0.j, 0.+0.j, 0.+0.j, 0.+0.j, 0.+0.j, 0.+0.j, 0.+0.j],
+                      dtype=complex64)
 
                 >>> s.sqrt_x(1, 2)
-                >>> list(s.state_vector())
-                [(0.5+0j), 0j, -0.5j, 0j, 0.5j, 0j, (0.5+0j), 0j]
+                >>> s.state_vector()
+                array([0.5+0.j , 0. +0.j , 0. -0.5j, 0. +0.j , 0. +0.5j, 0. +0.j ,
+                       0.5+0.j , 0. +0.j ], dtype=complex64)
         )DOC")
             .data());