Add tests

tests/deepdisc/conftest.py

@@ -10,3 +10,19 @@ def hsc_test_data_dir():
 @pytest.fixture
 def hsc_single_test_file(hsc_test_data_dir):
     return path.join(hsc_test_data_dir, "single_test.json")
+
+@pytest.fixture
+def hsc_triple_test_file(hsc_test_data_dir):
+    return [
+        path.join(hsc_test_data_dir, "G-10054-0,2-c1_scarlet_img.fits"),
+        path.join(hsc_test_data_dir, "I-10054-0,2-c1_scarlet_img.fits"),
+        path.join(hsc_test_data_dir, "R-10054-0,2-c1_scarlet_img.fits"),
+    ]
+
+@pytest.fixture
+def dc2_test_data_dir():
+    return path.join(TEST_DIR, "test_data/dc2")
+
+@pytest.fixture
+def dc2_single_test_file(dc2_test_data_dir):
+    return path.join(dc2_test_data_dir, "3828_2,2_12_images.npy")

tests/deepdisc/data_format/test_image_readers.py

@@ -0,0 +1,21 @@
+import os
+import pytest
+
+from deepdisc.data_format.image_readers import DC2ImageReader, HSCImageReader
+
+def test_read_hsc_data(hsc_triple_test_file):
+    """Test that we can read the test DC2 data."""
+    ir = HSCImageReader(norm="raw")
+    img = ir(hsc_triple_test_file)
+    assert img.shape[0] == 1050
+    assert img.shape[1] == 1025
+    assert img.shape[2] == 3
+
+
+def test_read_dc2_data(dc2_single_test_file):
+    """Test that we can read the test DC2 data."""
+    ir = DC2ImageReader(norm="raw")
+    img = ir(dc2_single_test_file)
+    assert img.shape[0] == 525
+    assert img.shape[1] == 525
+    assert img.shape[2] == 6