Система («пауза») в unittest

Question

У меня есть приложение Python, которое скомпилировано в .exe.
В некоторых случаях, программа завершается с sys.exit(), но не закрывать окно консоли Я делаю это с кодом ниже, который работает отлично:

try: 
    sys.exit() 
except SystemExit: 
    raise 
finally: 
    os.system("pause") 

Проблема заключается в том, что мне нужно проверить если мой код вызывает SystemExit, но из-за финального блока тесты не могут пройти - они просто бесконечны.
Есть ли возможность, чтобы убрать пропуск os.system("pause")?
Я могу добавить параметр по умолчанию, который предотвратит приостановку, но в моем случае это будет сложным - если есть какая-то возможность, я хотел бы сделать это в unittests.

Answer 1

Я бы просто высмеял os.system. Более того, даже sys.exit утверждать, если он вызывается.

# your_module.py 
import sys 
import os 


def something(): 
    try: 
     sys.exit() 
    except SystemExit: 
     raise 
    finally: 
     os.system("pause") 

И тест

import unittest 
import your_module 
try: 
    from unittest.mock import patch 
except ImportError: 
    from mock import patch 


class TestSysexit(unittest.TestCase): 

    @patch('your_module.os.system') 
    def test_system_exit(self, mock_os_system): 

     with self.assertRaises(SystemExit): 
      your_module.something() 
     mock_os_system.assert_called_once_with('pause') 

if __name__ == '__main__': 
    unittest.main() 

издеваться и патч в unittest.mock на Python3 и в ложном библиотеке python2.