small refactor

This commit is contained in:
Benoît Sierro
2023-10-04 08:21:37 +02:00
parent 5adde638ef
commit 27de20e6ca
2 changed files with 10 additions and 6 deletions

View File

@@ -184,7 +184,7 @@ def test_unique_name():
def test_propagation_series(tmp_path: Path):
params = Parameters(**PARAMS)
with pytest.raises(ValueError):
specs, props = propagation_series([])
specs, _ = propagation_series([])
flist = [tmp_path / f"prop{i}.zip" for i in range(10)]
for i, f in enumerate(flist):
@@ -196,5 +196,6 @@ def test_propagation_series(tmp_path: Path):
assert set(flist) == set(tmp_path.glob("*.zip"))
specs, propagations = propagation_series(flist)
assert specs.shape == (10, params.z_num, params.t_num)
assert all(prop.parameters.name == f"prop {i}" for i, prop in enumerate(propagations))