Пример #1
0
 def test_run(self):
     with patch.object(launch, 'MRJobLauncher') as m_launcher:
         cmd.main(args=['mrjob', 'run', 'script.py'])
         m_launcher.assert_called_once_with(
             args=['script.py'], from_cl=True)
Пример #2
0
 def _test_main_call(self, module, cmd_name):
     with patch.object(module, 'main') as m_main:
         cmd.main(args=['mrjob', cmd_name])
         m_main.assert_called_once_with([])