2020from cirq .circuits ._bucket_priority_queue import BucketPriorityQueue
2121
2222
23- def test_init ():
24- q = BucketPriorityQueue ()
23+ def test_init () -> None :
24+ q : BucketPriorityQueue = BucketPriorityQueue ()
2525 assert not q .drop_duplicate_entries
2626 assert list (q ) == []
2727 assert len (q ) == 0
@@ -42,7 +42,7 @@ def test_init():
4242 assert bool (q )
4343
4444
45- def test_eq ():
45+ def test_eq () -> None :
4646 eq = cirq .testing .EqualsTester ()
4747 eq .add_equality_group (
4848 BucketPriorityQueue (),
@@ -71,8 +71,8 @@ def test_eq():
7171 )
7272
7373
74- def test_enqueue_dequeue ():
75- q = BucketPriorityQueue ()
74+ def test_enqueue_dequeue () -> None :
75+ q : BucketPriorityQueue = BucketPriorityQueue ()
7676 q .enqueue (5 , 'a' )
7777 assert q == BucketPriorityQueue ([(5 , 'a' )])
7878 q .enqueue (4 , 'b' )
@@ -85,10 +85,10 @@ def test_enqueue_dequeue():
8585 _ = q .dequeue ()
8686
8787
88- def test_drop_duplicates_enqueue ():
89- q0 = BucketPriorityQueue ()
90- q1 = BucketPriorityQueue (drop_duplicate_entries = False )
91- q2 = BucketPriorityQueue (drop_duplicate_entries = True )
88+ def test_drop_duplicates_enqueue () -> None :
89+ q0 : BucketPriorityQueue = BucketPriorityQueue ()
90+ q1 : BucketPriorityQueue = BucketPriorityQueue (drop_duplicate_entries = False )
91+ q2 : BucketPriorityQueue = BucketPriorityQueue (drop_duplicate_entries = True )
9292 for q in [q0 , q1 , q2 ]:
9393 for _ in range (2 ):
9494 q .enqueue (0 , 'a' )
@@ -97,10 +97,10 @@ def test_drop_duplicates_enqueue():
9797 assert q2 == BucketPriorityQueue ([(0 , 'a' )], drop_duplicate_entries = True )
9898
9999
100- def test_drop_duplicates_dequeue ():
101- q0 = BucketPriorityQueue ()
102- q1 = BucketPriorityQueue (drop_duplicate_entries = False )
103- q2 = BucketPriorityQueue (drop_duplicate_entries = True )
100+ def test_drop_duplicates_dequeue () -> None :
101+ q0 : BucketPriorityQueue = BucketPriorityQueue ()
102+ q1 : BucketPriorityQueue = BucketPriorityQueue (drop_duplicate_entries = False )
103+ q2 : BucketPriorityQueue = BucketPriorityQueue (drop_duplicate_entries = True )
104104 for q in [q0 , q1 , q2 ]:
105105 q .enqueue (0 , 'a' )
106106 q .enqueue (0 , 'b' )
@@ -113,11 +113,11 @@ def test_drop_duplicates_dequeue():
113113 assert q2 == BucketPriorityQueue ([(0 , 'b' ), (0 , 'a' )], drop_duplicate_entries = True )
114114
115115
116- def test_same_priority_fifo ():
116+ def test_same_priority_fifo () -> None :
117117 a = (5 , 'a' )
118118 b = (5 , 'b' )
119119 for x , y in [(a , b ), (b , a )]:
120- q = BucketPriorityQueue ()
120+ q : BucketPriorityQueue = BucketPriorityQueue ()
121121 q .enqueue (* x )
122122 q .enqueue (* y )
123123 assert q
@@ -127,23 +127,23 @@ def test_same_priority_fifo():
127127 assert not q
128128
129129
130- def test_supports_arbitrary_offsets ():
130+ def test_supports_arbitrary_offsets () -> None :
131131 m = 1 << 60
132132
133- q_neg = BucketPriorityQueue ()
133+ q_neg : BucketPriorityQueue = BucketPriorityQueue ()
134134 q_neg .enqueue (- m + 0 , 'b' )
135135 q_neg .enqueue (- m - 4 , 'a' )
136136 q_neg .enqueue (- m + 4 , 'c' )
137137 assert list (q_neg ) == [(- m - 4 , 'a' ), (- m , 'b' ), (- m + 4 , 'c' )]
138138
139- q_pos = BucketPriorityQueue ()
139+ q_pos : BucketPriorityQueue = BucketPriorityQueue ()
140140 q_pos .enqueue (m + 0 , 'b' )
141141 q_pos .enqueue (m + 4 , 'c' )
142142 q_pos .enqueue (m - 4 , 'a' )
143143 assert list (q_pos ) == [(m - 4 , 'a' ), (m , 'b' ), (m + 4 , 'c' )]
144144
145145
146- def test_repr ():
146+ def test_repr () -> None :
147147 r = repr (BucketPriorityQueue (entries = [(1 , 2 ), (3 , 4 )], drop_duplicate_entries = True ))
148148 assert r .endswith ('BucketPriorityQueue(entries=[(1, 2), (3, 4)], drop_duplicate_entries=True)' )
149149
@@ -154,7 +154,7 @@ def test_repr():
154154 )
155155
156156
157- def test_str ():
157+ def test_str () -> None :
158158 s = str (BucketPriorityQueue (entries = [(1 , 2 ), (3 , 4 )], drop_duplicate_entries = True ))
159159 assert (
160160 s
0 commit comments