gh-91051: fix type watcher test to be robust to existing watcher (#107989)

This commit is contained in:
Carl Meyer 2023-08-16 13:13:32 -06:00 committed by GitHub
parent 8891a8821d
commit fce93c80ae
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -351,12 +351,10 @@ class TestTypeWatchers(unittest.TestCase):
self.clear_watcher(1)
def test_no_more_ids_available(self):
contexts = [self.watcher() for i in range(self.TYPE_MAX_WATCHERS)]
with ExitStack() as stack:
for ctx in contexts:
stack.enter_context(ctx)
with self.assertRaisesRegex(RuntimeError, r"no more type watcher IDs"):
self.add_watcher()
with self.assertRaisesRegex(RuntimeError, r"no more type watcher IDs"):
with ExitStack() as stack:
for _ in range(self.TYPE_MAX_WATCHERS + 1):
stack.enter_context(self.watcher())
class TestCodeObjectWatchers(unittest.TestCase):