diff --git a/tests/job/test_abci_job.py b/tests/job/test_abci_job.py index 40eea3d8..e8d190d5 100644 --- a/tests/job/test_abci_job.py +++ b/tests/job/test_abci_job.py @@ -40,7 +40,7 @@ def setUp(self) -> None: stderr_filename=self.stderr_filename, qsub_args=self.qsub_args, args=self.args, - tag=self.tag + tag=self.tag, ) def tearDown(self) -> None: @@ -59,9 +59,20 @@ def test_init(self) -> None: self.assertEqual( self.job.cmd, [ - "qsub", "-g", self.job_group, "-o", str(self.stdout_filename), "-e", str(self.stderr_filename), - "-N", self.job_name, "-l", str(self.job_filename), "arg1", "arg2" - ] + "qsub", + "-g", + self.job_group, + "-o", + str(self.stdout_filename), + "-e", + str(self.stderr_filename), + "-N", + self.job_name, + "-l", + str(self.job_filename), + "arg1", + "arg2", + ], ) def test_submit(self) -> None: diff --git a/tests/job/test_abci_job_manager.py b/tests/job/test_abci_job_manager.py index b3d1bb4c..350d40ef 100644 --- a/tests/job/test_abci_job_manager.py +++ b/tests/job/test_abci_job_manager.py @@ -31,7 +31,7 @@ def setUp(self) -> None: self.job_group, job_name=self.job_name, work_dir=self.work_dir, - n_max_jobs=self.n_max_jobs + n_max_jobs=self.n_max_jobs, ) def tearDown(self) -> None: @@ -65,9 +65,8 @@ def test_available_slots_full(self) -> None: with patch("subprocess.run", return_value=qstat_xml("tests/job/qstat_dat.txt")): result = self.executor.available_slots() - num_available_slots = ( - self.n_max_jobs - - len([j for j in self.executor.job_list if j.status == JobStatus.RUNNING or JobStatus.WAITING]) + num_available_slots = self.n_max_jobs - len( + [j for j in self.executor.job_list if j.status == JobStatus.RUNNING or JobStatus.WAITING] ) self.assertEqual(result, num_available_slots) @@ -96,9 +95,8 @@ def test_available_slots_pending(self) -> None: with patch("subprocess.run", return_value=qstat_xml("tests/job/qstat_dat 2r_1qw.txt")): result = self.executor.available_slots() - num_available_slots = ( - self.n_max_jobs - - len([j for j in self.executor.job_list if j.status == JobStatus.RUNNING or JobStatus.WAITING]) + num_available_slots = self.n_max_jobs - len( + [j for j in self.executor.job_list if j.status == JobStatus.RUNNING or JobStatus.WAITING] ) self.assertEqual(result, num_available_slots)