Update test_widget.py

COSC-Lab · Jun 19, 2024 · 94aaaad · 94aaaad
1 parent 4d0e2c8
commit 94aaaad
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/src/napari_fast4dreg/_tests/test_widget.py b/src/napari_fast4dreg/_tests/test_widget.py
@@ -1,14 +1,17 @@
 import numpy as np
 import tifffile 
+import napari 
 
-from napari_fast4dreg._widget import Fast4DReg_widget
+from napari_fast4dreg._widget import Fast4DReg_widget 
 
 def test_Fast4DReg_widget():
     # working path is napari-fast4dreg/napari-fast4dreg/
     # 
-
+    viewer = napari.Viewer()
     test_image = tifffile.imread('example_files/xtitched_organoid_timelapse_ch0_cytosol_ch1_nuclei.tif')
     reference_registered_image = tifffile.imread('example_files/registered.tif')
+    viewer.add_image(reference_registered_image)
+
     out_folder = 'example_files/'
     resgistered_image = Fast4DReg_widget(image = test_image,
                                         axes = 1, 
@@ -20,4 +23,7 @@ def test_Fast4DReg_widget():
                                         crop_output = True, 
                                         export_csv = True )
 
-    assert resgistered_image == reference_registered_image
+    # get layer data 
+    layers = v.layers()
+
+    assert layers[0].data == layers[-1].data