Skip to content

Commit b1b7c81

Browse files
committed
Update test_plugin_inference.py
1 parent 6bb71bb commit b1b7c81

File tree

1 file changed

+5
-9
lines changed

1 file changed

+5
-9
lines changed

napari_cellseg3d/_tests/test_plugin_inference.py

Lines changed: 5 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,3 @@
1-
from pathlib import Path
2-
3-
from tifffile import imread
4-
51
from napari_cellseg3d._tests.fixtures import LogFixture
62
from napari_cellseg3d.code_models.instance_segmentation import (
73
INSTANCE_SEGMENTATION_METHOD_LIST,
@@ -17,10 +13,9 @@
1713

1814

1915
def test_inference(make_napari_viewer_proxy, qtbot):
20-
im_path = str(Path(__file__).resolve().parent / "res/test.tif")
21-
image = imread(im_path)
22-
23-
assert image.shape == (6, 6, 6)
16+
dims = 6
17+
image = rand_gen.random(size=(dims, dims, dims))
18+
# assert image.shape == (dims, dims, dims)
2419

2520
viewer = make_napari_viewer_proxy()
2621
widget = Inferer(viewer)
@@ -46,14 +41,15 @@ def test_inference(make_napari_viewer_proxy, qtbot):
4641
widget.model_choice.addItem(test_model_name)
4742
widget.model_choice.setCurrentText(test_model_name)
4843

44+
widget.window_infer_box.setChecked(False)
4945
widget.worker_config = widget._set_worker_config()
5046
assert widget.worker_config is not None
5147
assert widget.model_info is not None
52-
widget.window_infer_box.setChecked(False)
5348

5449
worker = widget._create_worker_from_config(widget.worker_config)
5550
assert worker.config is not None
5651
assert worker.config.model_info is not None
52+
assert worker.config.sliding_window_config.is_enabled() is False
5753
worker.config.layer = viewer.layers[0].data
5854
worker.config.post_process_config.instance.enabled = True
5955
worker.config.post_process_config.instance.method = (

0 commit comments

Comments
 (0)