Skip to content

Commit 88f918d

Browse files
committed
Update test typing
1 parent ca118d7 commit 88f918d

File tree

1 file changed

+18
-2
lines changed

1 file changed

+18
-2
lines changed

Lib/test/test_queue.py

Lines changed: 18 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -10,6 +10,16 @@
1010
from test.support import import_helper
1111
from test.support import threading_helper
1212

13+
import typing as t
14+
15+
if t.TYPE_CHECKING:
16+
import queue
17+
18+
parent_class = unittest.TestCase
19+
20+
else:
21+
parent_class = object
22+
1323
# queue module depends on threading primitives
1424
threading_helper.requires_working_threading(module=True)
1525

@@ -55,7 +65,10 @@ def run(self):
5565
# is supposed to raise an exception, call do_exceptional_blocking_test()
5666
# instead.
5767

58-
class BlockingTestMixin:
68+
class BlockingTestMixin(parent_class):
69+
if t.TYPE_CHECKING:
70+
queue = queue
71+
type2test: t.Type[queue.Queue]
5972

6073
def do_blocking_test(self, block_func, block_args, trigger_func, trigger_args):
6174
thread = _TriggerThread(trigger_func, trigger_args)
@@ -797,7 +810,10 @@ class CFailingQueueTest(FailingQueueTest, unittest.TestCase):
797810
queue = c_queue
798811

799812

800-
class BaseSimpleQueueTest:
813+
class BaseSimpleQueueTest(parent_class):
814+
if t.TYPE_CHECKING:
815+
queue = queue
816+
type2test: t.Type[queue.Queue]
801817

802818
def setUp(self):
803819
self.q = self.type2test()

0 commit comments

Comments
 (0)