Skip to content

Commit d35da41

Browse files
committed
WNet eval test
1 parent c1aecb8 commit d35da41

File tree

2 files changed

+223
-218
lines changed

2 files changed

+223
-218
lines changed

napari_cellseg3d/_tests/test_training.py

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -92,3 +92,15 @@ def test_unsupervised_training(make_napari_viewer_proxy):
9292
)
9393
)
9494
assert isinstance(res, TrainingReport)
95+
assert not res.show_plot
96+
widget.worker.config.eval_volume_dict = [
97+
{"image": im_path_str, "label": im_path_str}
98+
]
99+
widget.worker._get_data()
100+
eval_res = widget.worker._eval(
101+
model=WNetFixture(),
102+
epoch=-10,
103+
)
104+
assert isinstance(eval_res, TrainingReport)
105+
assert eval_res.show_plot
106+
assert eval_res.epoch == -10

0 commit comments

Comments
 (0)