Skip to content

Commit b2930fe

Browse files
committed
Sqgturb generate produces exactly n_steps now, not n_steps+1, in order to match other generators
1 parent 6eb5aeb commit b2930fe

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

tests/data_sqgturb_test.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -40,5 +40,5 @@ def test_variable_sizes(sqgturb):
4040
traj = sqgturb.generate(n_steps=n_steps)
4141

4242
assert traj.system_dim == 18432
43-
assert traj.sizes['time'] == n_steps+1
44-
assert traj.dab.flatten().shape == (n_steps+1, 18432)
43+
assert traj.sizes['time'] == n_steps
44+
assert traj.dab.flatten().shape == (n_steps, 18432)

0 commit comments

Comments
 (0)