Skip to content

Commit 3229fba

Browse files
committed
Add test for getting multiple init conds
1 parent 7948e37 commit 3229fba

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

paths_cli/tests/test_parameters.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -234,6 +234,14 @@ def test_get_none(self, num_in_file):
234234
obj = INIT_CONDS.get(st, None)
235235
assert obj == stored_things[num_in_file - 1]
236236

237+
def test_get_multiple(self):
238+
filename = self.create_file('number-traj')
239+
storage = paths.Storage(filename, mode='r')
240+
traj0, traj1 = self.PARAMETER.get(storage, (0, 1))
241+
assert traj0 == self.traj
242+
assert traj1 == self.other_traj
243+
244+
237245
class TestINIT_SNAP(ParamInstanceTest):
238246
PARAMETER = INIT_SNAP
239247
def setup(self):

0 commit comments

Comments
 (0)