Added unit test for new signature. this refs #294

ornlneutronimaging · Mar 2, 2024 · bab5447 · bab5447
1 parent 064267d
commit bab5447
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/tests/unit/backend/dataio/test_data.py b/tests/unit/backend/dataio/test_data.py
@@ -115,6 +115,9 @@ def test_load_data(
     # case_1: load data from given directory
     rst = load_data(ct_dir="/tmp", ob_dir="/tmp", dc_dir="/tmp")
     np.testing.assert_almost_equal(np.array(rst).flatten(), np.arange(1, 5, dtype=float))
+    # case_2: load ct from directory, ob and dc from files
+    rst = load_data(ct_dir="/tmp", ob_files=["3", "4"], dc_files=["5", "6"])
+    np.testing.assert_almost_equal(np.array(rst).flatten(), np.arange(1, 5, dtype=float))
 
 
 def test_forgiving_reader():