Use a contextmanager to setUp a test case.
(testcase: unittest.TestCase, cm: Any)
| 787 | |
| 788 | # From https://nedbatchelder.com/blog/201508/using_context_managers_in_test_setup.html |
| 789 | def setup_with_context_manager(testcase: unittest.TestCase, cm: Any) -> Any: |
| 790 | """Use a contextmanager to setUp a test case.""" |
| 791 | val = cm.__enter__() |
| 792 | testcase.addCleanup(cm.__exit__, None, None, None) |
| 793 | return val |
| 794 | |
| 795 | |
| 796 | def main(**kwargs: Any) -> None: |