@@ -39,16 +39,13 @@ def testInstallHandler(self):
|
39 | 39 | |
40 | 40 | def testRegisterResult(self): |
41 | 41 | result = unittest.TestResult() |
42 | | -unittest.registerResult(result) |
43 | | - |
44 | | -for ref in unittest.signals._results: |
45 | | -if ref is result: |
46 | | -break |
47 | | -elif ref is not result: |
48 | | -self.fail("odd object in result set") |
49 | | -else: |
50 | | -self.fail("result not found") |
| 42 | +self.assertNotIn(result, unittest.signals._results) |
51 | 43 | |
| 44 | +unittest.registerResult(result) |
| 45 | +try: |
| 46 | +self.assertIn(result, unittest.signals._results) |
| 47 | +finally: |
| 48 | +unittest.removeResult(result) |
52 | 49 | |
53 | 50 | def testInterruptCaught(self): |
54 | 51 | default_handler = signal.getsignal(signal.SIGINT) |
|