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)