From c595ce28eee1a228aa22ea7c89c72e33b880f7e9 Mon Sep 17 00:00:00 2001 From: Lourens Veen Date: Sun, 1 Jun 2025 13:48:38 +0200 Subject: [PATCH] Fix parallel stellar evo test broken by 279a06920 --- .../ext_tests/test_parallel_stellar_evolution.py | 13 +++++++++---- 1 file changed, 9 insertions(+), 4 deletions(-) diff --git a/src/tests/ext_tests/test_parallel_stellar_evolution.py b/src/tests/ext_tests/test_parallel_stellar_evolution.py index 7bfe8b08fe..c4bd766da1 100644 --- a/src/tests/ext_tests/test_parallel_stellar_evolution.py +++ b/src/tests/ext_tests/test_parallel_stellar_evolution.py @@ -81,15 +81,20 @@ def test4(self): def test5(self): print("Testing ParallelStellarEvolution individual options") + n_workers = 2 base_name = os.path.join(get_path_to_results(), "parallel_stellar_evolution_out_") - for filename in [base_name+str(i) for i in range(3)]: + for filename in [base_name+str(i) for i in range(n_workers)]: if os.path.exists(filename): os.remove(filename) - parallel = ParallelStellarEvolution(self.code_factory, number_of_workers=2, - individual_options=[dict(redirect_file=base_name+str(i)) for i in range(3)], redirection="file", **default_options) + parallel = ParallelStellarEvolution( + self.code_factory, number_of_workers=n_workers, + individual_options=[ + dict(redirect_file=base_name+str(i)) + for i in range(n_workers)], + redirection="file", **default_options) - for filename in [base_name+str(i) for i in range(3)]: + for filename in [base_name+str(i) for i in range(n_workers)]: self.assertTrue(os.path.exists(filename)) parallel.stop()