bpo-30813: Fix unittest when hunting refleaks (#2502) (#2506) · python/cpython@714afcc (original) (raw)

Original file line number Diff line number Diff line change
@@ -528,6 +528,9 @@ def _get_module_from_name(name):
528 528 pickle.loads(pickle.dumps(test, proto))
529 529
530 530 def test_discover_with_module_that_raises_SkipTest_on_import(self):
531 +if not unittest.BaseTestSuite._cleanup:
532 +raise unittest.SkipTest("Suite cleanup is disabled")
533 +
531 534 loader = unittest.TestLoader()
532 535
533 536 def _get_module_from_name(name):
@@ -548,6 +551,9 @@ def _get_module_from_name(name):
548 551 pickle.loads(pickle.dumps(suite, proto))
549 552
550 553 def test_discover_with_init_module_that_raises_SkipTest_on_import(self):
554 +if not unittest.BaseTestSuite._cleanup:
555 +raise unittest.SkipTest("Suite cleanup is disabled")
556 +
551 557 vfs = {abspath('/foo'): ['my_package'],
552 558 abspath('/foo/my_package'): ['__init__.py', 'test_module.py']}
553 559 self.setup_import_issue_package_tests(vfs)