def test_submit_pops_global_state(self):
submit()
self.mock_pop_state.assert_called()
def test_submit_pops_global_state_even_when_an_error_happens(self):
self.mock_submit.side_effect = self._die
with self.assertRaises(Exception):
submit()
self.mock_pop_state.assert_called()
def test_submit_passes_stream_job_logs_option(self):
submit(stream_job_logs=self.random_parameter)
self.assertEqual(self.random_parameter,
self._arguments.stream_job_logs)
def test_submit_passes_command_option(self):
submit(command=self.random_parameter)
self.assertEqual(self.random_parameter, self._arguments.command)
def test_submit_passes_ram(self):
submit(ram=self.random_parameter)
self.assertEqual(self.random_parameter, self._arguments.ram)
def test_submit_passes_num_gpus(self):
submit(num_gpus=self.random_parameter)
self.assertEqual(self.random_parameter, self._arguments.num_gpus)
def test_submit_passes_params(self):
submit(params=self.random_parameter)
self.assertEqual(self.random_parameter, self._arguments.params)
def test_submit_passes_entrypoint(self):
submit(entrypoint=self.random_parameter)
self.assertEqual(self.random_parameter, self._arguments.entrypoint)
def test_submit_passes_project_name(self):
submit(project_name=self.random_parameter)
self.assertEqual(self.random_parameter, self._arguments.project_name)
def test_submit_passes_job_dir(self):
submit(job_directory=self.random_parameter)
self.assertEqual(self.random_parameter, self._arguments.job_directory)
def test_submit_passes_scheduler_config_name(self):
submit(scheduler_config=self.random_parameter)
self.assertEqual(self.random_parameter,
self._arguments.scheduler_config)
