Skip to content

Commit 10361f7

Browse files
committed
Add runset config filename tests
1 parent ea73385 commit 10361f7

File tree

1 file changed

+9
-0
lines changed

1 file changed

+9
-0
lines changed

test/test_runset.py

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -30,6 +30,7 @@ def test_check_repr() -> None:
3030
assert 'csv_file' in repr(runset)
3131
assert 'console_msgs' in repr(runset)
3232
assert 'diagnostics_file' not in repr(runset)
33+
assert 'config_file' in repr(runset)
3334

3435

3536
def test_check_retcodes() -> None:
@@ -106,6 +107,11 @@ def test_output_filenames_one_proc_per_chain() -> None:
106107
stdout_file.endswith(f"_stdout_{id}.txt")
107108
for id, stdout_file in zip(chain_ids, runset.stdout_files)
108109
)
110+
assert len(runset.config_files) == len(chain_ids)
111+
assert all(
112+
config_file.endswith(f"_{id}_config.json")
113+
for id, config_file in zip(chain_ids, runset.config_files)
114+
)
109115

110116
cmdstan_args_other_files = CmdStanArgs(
111117
model_name='bernoulli',
@@ -153,6 +159,8 @@ def test_output_filenames_threading() -> None:
153159
)
154160
assert len(runset.stdout_files) == 1
155161
assert runset.stdout_files[0].endswith("_stdout.txt")
162+
assert len(runset.config_files) == 1
163+
assert runset.config_files[0].endswith("_config.json")
156164

157165
cmdstan_args_other_files = CmdStanArgs(
158166
model_name='bernoulli',
@@ -198,6 +206,7 @@ def test_output_filenames_single_chain() -> None:
198206
runset = RunSet(args=cmdstan_args, chains=1, one_process_per_chain=True)
199207
base_file = runset._base_outfile
200208
assert runset.stdout_files[0].endswith(f"{base_file}_stdout.txt")
209+
assert runset.config_files[0].endswith(f"{base_file}_config.json")
201210

202211
cmdstan_args_other_files = CmdStanArgs(
203212
model_name='bernoulli',

0 commit comments

Comments
 (0)