diff --git a/tests/test_checkchmjob.py b/tests/test_checkchmjob.py index a0d902b..5f0e983 100644 --- a/tests/test_checkchmjob.py +++ b/tests/test_checkchmjob.py @@ -251,5 +251,6 @@ def test_check_chm_job_all_tasks_complete_with_submit(self): finally: shutil.rmtree(temp_dir) + if __name__ == '__main__': unittest.main() diff --git a/tests/test_chmjobcreator.py b/tests/test_chmjobcreator.py index 5c1fddd..7dde1e1 100644 --- a/tests/test_chmjobcreator.py +++ b/tests/test_chmjobcreator.py @@ -384,7 +384,7 @@ def test_create_job_three_images_five_tiles_per_job(self): 7) o_img_list = [] - for x in range(1,8): + for x in range(1, 8): o_img = config.get(str(x), CHMJobCreator.CONFIG_OUTPUT_IMAGE) o_img_list.append(o_img) diff --git a/tests/test_clusterfactory.py b/tests/test_clusterfactory.py index befd98d..ff25124 100644 --- a/tests/test_clusterfactory.py +++ b/tests/test_clusterfactory.py @@ -46,5 +46,6 @@ def test_get_cluster_by_name(self): c = cfac.get_cluster_by_name(CometCluster.CLUSTER) self.assertEqual(c.get_cluster(), CometCluster.CLUSTER) + if __name__ == '__main__': unittest.main() diff --git a/tests/test_taskstats.py b/tests/test_taskstats.py index 15cd3f5..f837ac8 100644 --- a/tests/test_taskstats.py +++ b/tests/test_taskstats.py @@ -46,5 +46,6 @@ def test_getter_and_setters(self): self.assertEqual(ts.get_max_memory_in_kb(), 300) self.assertEqual(ts.get_total_memory_in_kb(), 400) + if __name__ == '__main__': unittest.main() diff --git a/tests/test_tasksummaryfactory.py b/tests/test_tasksummaryfactory.py index fa1a8f0..aab3336 100644 --- a/tests/test_tasksummaryfactory.py +++ b/tests/test_tasksummaryfactory.py @@ -493,5 +493,6 @@ def test_get_task_summary(self): '\nMerge tasks: 0% complete (0 of ' '1 completed)\n') + if __name__ == '__main__': unittest.main()