Tests

CamaraLab · Mar 11, 2024 · c7e0fd1 · c7e0fd1
1 parent ed8b7c6
commit c7e0fd1
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/tests/test_qgw.py b/tests/test_qgw.py
@@ -1,13 +1,18 @@
-from cajal.qgw import slb_parallel, combined_slb_quantized_gw
+from cajal.qgw import slb_parallel
+from cajal.combined_slb_qgw import combined_slb_quantized_gw
 import os
 
-def test():
+
+def test_slb_parallel():
     slb_parallel(
         intracell_csv_loc="tests/icdm.csv",
         out_csv="tests/gw1.csv",
         num_processes=2,
     )
     os.remove("tests/gw1.csv")
+
+
+def test_combined_slb_quantized_gw():
     combined_slb_quantized_gw(
         input_icdm_csv_location="tests/icdm.csv",
         gw_out_csv_location="tests/slb_qgw.csv",