Дразнящий __main__

Question

Я хотел бы обеспечить с тестами, которые: - приложение может не быть импортированы - приложение может быть запущен в качестве реального приложения (например: питон src.py)

Я заинтересован о том, что , почему следующее не работает:

src.py

class A: 
def x(self): 
    print('this is x') 

if __name__ == '__main__': 
    A().x() 

test.py (фрагмент)

class Test(unittest.TestCase): 
    @mock.patch('src.A.x') 
    def test_main(self, mock_x): 
    import src 
    mock_x.assert_any_call() 

Этот тест не подходит ... почему?

Answer 1

Поскольку имя модуля при импорте составляет src, а не __main__.

Самым простым решением было бы перенести этот код в функцию:

def main(): 
    A().x() 

if __name__ == '__main__': 
    main() 

и в тесте, вы бы вызвать src.main()

@mock.patch('src.A.x') 
def test_main(self, mock_x): 
    import src 
    src.main() 
    mock_x.assert_any_call() 

Answer 2

Чтобы проверить, что модуль не ввозу вы делаете не нужно использовать насмешки.

См. assertRaises.

Просто проверьте, не возникает ли ошибка при импорте od модуля.

with self.assertRaises(...): 
    ...