def test_main_html(self, mock_sys, mock_trace_module, mock_reports): mock_sys.argv = ["profimp", "import re", "--html"] mock_trace_module.return_value main.main() mock_trace_module.assert_called_once_with("import re") mock_reports.to_html.assert_called_once_with( mock_trace_module.return_value)
def test_main_too_few_args(self, mock_sys, mock_print_help): mock_sys.argv = ["profimp"] main.main() mock_print_help.assert_called_once_with()