Skip to content

Commit f85ef4f

Browse files
committed
add very simple basic test
1 parent 32f11ab commit f85ef4f

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

tests/test_validation_and_test_split.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -31,3 +31,9 @@ def test_select_spectra_belonging_to_inchikey(sample_spectra):
3131
result = select_spectra_belonging_to_inchikey(sample_spectra, inchikeys)
3232
assert len(result) == 3
3333
assert result[0].get("inchikey") == 14 * "A"
34+
35+
36+
def test_split_spectra_in_random_inchikey_sets(sample_spectra):
37+
# TODO: this is still a dummy test mostly
38+
val, test, train = split_spectra_in_random_inchikey_sets(sample_spectra, 2)
39+
assert len(val) + len(test) + len(train) == 4

0 commit comments

Comments
 (0)