Skip to content

Commit 086b03e

Browse files
authored
Fix more typing issues (#7388)
- This is the result of adding "-> None" to a bunch of tests and fixing all the issues that result.
1 parent 09b44c0 commit 086b03e

66 files changed

Lines changed: 958 additions & 864 deletions

File tree

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

cirq-core/cirq/circuits/_bucket_priority_queue_test.py

Lines changed: 20 additions & 20 deletions
Original file line numberDiff line numberDiff line change
@@ -20,8 +20,8 @@
2020
from 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

Comments
 (0)