Skip to content

Commit 29ddbfd

Browse files
committed
Fix metric output filenames test to reflect one proc per chain
1 parent b138308 commit 29ddbfd

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

test/test_runset.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -331,6 +331,13 @@ def test_metric_output_filename() -> None:
331331
)
332332
runset = RunSet(args=cmdstan_args, chains=2)
333333
assert len(runset.metric_files) == 2
334+
assert all(
335+
f.endswith(f"{base_file}_{i}_metric.json")
336+
for i, f in zip(chain_ids, runset.metric_files)
337+
)
338+
339+
runset = RunSet(args=cmdstan_args, chains=2, one_process_per_chain=False)
340+
assert len(runset.metric_files) == 2
334341
assert all(
335342
f.endswith(f"{base_file}_metric_{i}.json")
336343
for i, f in zip(chain_ids, runset.metric_files)

0 commit comments

Comments
 (0)