add one (1) unittest

.gitignore

@@ -156,4 +156,4 @@ _html/
 results/
 
 # Test data
-data/
+/data/

tests/kbmod_ml/conftest.py

@@ -0,0 +1,18 @@
+import os
+import pytest
+
+from fibad import Fibad
+
+@pytest.fixture
+def test_data_path():
+    test_dir = os.path.dirname(__file__)
+    return os.path.join(test_dir, "data/")
+
+@pytest.fixture
+def basic_fibad_instance(test_data_path):
+    config_path = os.path.join(test_data_path, "basic_test_config.toml")
+    bfi = Fibad(config_file=config_path)
+
+    bfi.config["general"]["data_dir"] = test_data_path
+
+    return bfi

tests/kbmod_ml/data/basic_test_config.toml

@@ -0,0 +1,18 @@
+[general]
+dev_mode = true
+
+data_dir = "."
+
+results_dir = "./results"
+
+[model]
+name = "kbmod_ml.models.cnn.CNN"
+
+weights_filepath = "resnet50.pth"
+
+[infer]
+chromadb = false
+
+[kbmod_ml]
+true_positive_file_name = 'true_positive_toy.npy'
+false_positive_file_name = 'false_positive_toy.npy'

tests/kbmod_ml/test_kbmod_stamps.py

@@ -0,0 +1,11 @@
+from fibad import Fibad
+from kbmod_ml.data_sets.kbmod_stamps import KbmodStamps
+import os
+
+def test_basic_loading(basic_fibad_instance):
+    ds = KbmodStamps(config=basic_fibad_instance.config, split="train")
+
+    test_tensor, label = ds[0]
+
+    assert test_tensor.shape == (1, 21, 21)
+    assert label == 0 or label == 1