@@ -31,8 +31,8 @@ def test_preprocessing_pipeline():
3131 tests_dir = get_tests_dir ()
3232 npy_file = join (tests_dir , "inputs/21_5.npy" )
3333 npy_mask = join (tests_dir , "inputs/21_5_mask.npy" )
34- out_zarr = join (tests_dir , "inputs/21_5 .zarr" )
35- out_pkl = join (tests_dir , "inputs/21_5_mask .pkl" )
34+ out_zarr = join (tests_dir , "output_zarr .zarr" )
35+ out_pkl = join (tests_dir , "output .pkl" )
3636
3737 utils .run_preprocessing_pipeline (
3838 npy_file , npy_mask = npy_mask , out_zarr = out_zarr , out_pkl = out_pkl
@@ -45,7 +45,6 @@ def test_preprocessing_pipeline():
4545 from numpy import load as load_numpy
4646
4747 img = zarr_to_da (open_zarr (out_zarr )).compute ()
48-
4948 assert array_equal (img , load_numpy (npy_file ))
5049
5150 def capture (command ):
@@ -58,10 +57,11 @@ def capture(command):
5857 out , err = proc .communicate ()
5958 return out , err , proc .returncode
6059
60+ odb = join (tests_dir , "patch_information.db" )
6161 command = [
6262 "poetry" , "run" , "pathflowai-preprocess" ,
6363 "preprocess-pipeline" ,
64- "-odb" , "patch_information.db" ,
64+ "-odb" , odb ,
6565 "--preprocess" ,
6666 "--patches" ,
6767 "--basename" , "21_5" ,
@@ -72,4 +72,7 @@ def capture(command):
7272 "-t" , "0.05"
7373 ]
7474 out , err , exitcode = capture (command )
75+ assert exists (out_zarr )
76+ assert exists (out_pkl )
77+ assert exists (odb )
7578 assert exitcode == 0
0 commit comments