modify the unit test to assert the full circuit

quantumlib · Mar 23, 2024 · db4a085 · db4a085
1 parent 955b79e
commit db4a085
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/glue/cirq/stimcirq/_cirq_to_stim_test.py b/glue/cirq/stimcirq/_cirq_to_stim_test.py
@@ -352,7 +352,14 @@ def test_loop_with_one_rep():
         )
     )
     sc = stimcirq.cirq_circuit_to_stim_circuit(cc)
-    assert str(sc).splitlines().count('TICK') > 1
+    assert str(sc) == """H 0
+TICK
+H 0
+TICK
+H 0
+TICK
+H 0
+TICK"""
 
 def test_on_tagged_loop():
     a, b = cirq.LineQubit.range(2)