Skip to content
Snippets Groups Projects

Test/add xfel calibrate cli tests

Merged Robert Rosca requested to merge test/add-xfel-calibrate-cli-tests into master
1 file
+ 34
19
Compare changes
  • Side-by-side
  • Inline
+ 34
19
@@ -4,32 +4,47 @@ from xfel_calibrate.calibrate import balance_sequences
def test_balance_sequences():
ret = balance_sequences(in_folder="/gpfs/exfel/exp/CALLAB/202031/p900113/raw", # noqa
run=9992, sequences=[0, 2, 5, 10, 20, 50, 100],
sequences_per_node=1, karabo_da=["all"],
max_nodes=8)
ret = balance_sequences(
in_folder="/gpfs/exfel/exp/CALLAB/202031/p900113/raw", # noqa
run=9992,
sequences=[0, 2, 5, 10, 20, 50, 100],
sequences_per_node=1,
karabo_da=["all"],
max_nodes=8,
)
expected = [[0], [2]]
assert expected == ret
ret = balance_sequences(in_folder="/gpfs/exfel/exp/CALLAB/202031/p900113/raw", # noqa
run=9992, sequences=[-1],
sequences_per_node=1, karabo_da=["JNGFR01"],
max_nodes=3)
ret = balance_sequences(
in_folder="/gpfs/exfel/exp/CALLAB/202031/p900113/raw", # noqa
run=9992,
sequences=[-1],
sequences_per_node=1,
karabo_da=["JNGFR01"],
max_nodes=3,
)
expected = []
assert expected == ret
with pytest.raises(ValueError) as e:
balance_sequences(in_folder="/gpfs/exfel/exp/CALLAB/202031/p900113/raw", # noqa
run=9992, sequences=[1991, 2021],
sequences_per_node=1, karabo_da=["all"],
max_nodes=3)
assert 'Selected sequences [1991, 2021]]' in e.value()
balance_sequences(
in_folder="/gpfs/exfel/exp/CALLAB/202031/p900113/raw", # noqa
run=9992,
sequences=[1991, 2021],
sequences_per_node=1,
karabo_da=["all"],
max_nodes=3,
)
assert "Selected sequences [1991, 2021]]" in e.value()
with pytest.raises(ValueError) as e:
balance_sequences(in_folder="/gpfs/exfel/exp/CALLAB/202031/p900113/raw", # noqa
run=9992, sequences=[1991, 2021],
sequences_per_node=1, karabo_da=-1,
max_nodes=3)
assert 'karabo_da as a string or list' in e.value()
balance_sequences(
in_folder="/gpfs/exfel/exp/CALLAB/202031/p900113/raw", # noqa
run=9992,
sequences=[1991, 2021],
sequences_per_node=1,
karabo_da=-1,
max_nodes=3,
)
assert "karabo_da as a string or list" in e.value()
Loading