add very simple basic test

matchms · Aug 18, 2023 · f85ef4f · f85ef4f
1 parent 32f11ab
commit f85ef4f
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/tests/test_validation_and_test_split.py b/tests/test_validation_and_test_split.py
@@ -31,3 +31,9 @@ def test_select_spectra_belonging_to_inchikey(sample_spectra):
     result = select_spectra_belonging_to_inchikey(sample_spectra, inchikeys)
     assert len(result) == 3
     assert result[0].get("inchikey") == 14 * "A"
+
+
+def test_split_spectra_in_random_inchikey_sets(sample_spectra):
+    # TODO: this is still a dummy test mostly
+    val, test, train = split_spectra_in_random_inchikey_sets(sample_spectra, 2)
+    assert len(val) + len(test) + len(train) == 4