diff --git a/Lib/test/test_typing.py b/Lib/test/test_typing.py index a3a6b4e2274aa3..d68c3259699598 100644 --- a/Lib/test/test_typing.py +++ b/Lib/test/test_typing.py @@ -4391,19 +4391,19 @@ def blah(): blah() - @patch("typing._overload_registry", - defaultdict(lambda: defaultdict(dict))) def test_overload_on_compiled_functions(self): - # The registry starts out empty: - self.assertEqual(typing._overload_registry, {}) - - # This should just not fail: - overload(sum) - overload(print) - - # No overloads are recorded (but, it still has a side-effect): - self.assertEqual(typing.get_overloads(sum), []) - self.assertEqual(typing.get_overloads(print), []) + with patch("typing._overload_registry", + defaultdict(lambda: defaultdict(dict))): + # The registry starts out empty: + self.assertEqual(typing._overload_registry, {}) + + # This should just not fail: + overload(sum) + overload(print) + + # No overloads are recorded (but, it still has a side-effect): + self.assertEqual(typing.get_overloads(sum), []) + self.assertEqual(typing.get_overloads(print), []) def set_up_overloads(self): def blah():