def test_new_labels(qtbot):
"""Test adding new labels layer."""
# Add labels to empty viewer
viewer = ViewerModel()
view = QtViewer(viewer)
qtbot.addWidget(view)
viewer._new_labels()
assert np.max(viewer.layers[0].data) == 0
assert len(viewer.layers) == 1
assert view.layers.vbox_layout.count() == 2 * len(viewer.layers) + 2
assert viewer.dims.ndim == 2
assert view.dims.nsliders == viewer.dims.ndim
assert np.sum(view.dims._displayed_sliders) == 0
# Add labels with image already present
viewer = ViewerModel()
view = QtViewer(viewer)
qtbot.addWidget(view)
np.random.seed(0)
data = np.random.random((10, 15))
viewer.add_image(data)
viewer._new_labels()
assert np.max(viewer.layers[1].data) == 0
assert len(viewer.layers) == 2
assert view.layers.vbox_layout.count() == 2 * len(viewer.layers) + 2
assert viewer.dims.ndim == 2
assert view.dims.nsliders == viewer.dims.ndim
assert np.sum(view.dims._displayed_sliders) == 0
view.shutdown()
def test_new_labels_empty():
"""Test adding new labels layer to empty viewer."""
viewer = ViewerModel()
viewer._new_labels()
assert len(viewer.layers) == 1
assert np.max(viewer.layers[0].data) == 0
assert viewer.dims.ndim == 2
# Default shape when no data is present is 512x512
np.testing.assert_equal(viewer.layers[0].data.shape, (512, 512))
def test_new_labels_scaled_translated_image():
"""Test adding new labels layer with transformed image present."""
viewer = ViewerModel()
np.random.seed(0)
data = np.random.random((10, 15))
viewer.add_image(data, scale=(3, 3), translate=(20, -5))
viewer._new_labels()
assert len(viewer.layers) == 2
assert np.max(viewer.layers[1].data) == 0
assert viewer.dims.ndim == 2
np.testing.assert_almost_equal(viewer.layers[1].data.shape, (10, 15))
np.testing.assert_almost_equal(viewer.layers[1].scale, (3, 3))
np.testing.assert_almost_equal(viewer.layers[1].translate, (20, -5))
def test_new_labels_image():
"""Test adding new labels layer with image present."""
viewer = ViewerModel()
np.random.seed(0)
data = np.random.random((10, 15))
viewer.add_image(data)
viewer._new_labels()
assert len(viewer.layers) == 2
assert np.max(viewer.layers[1].data) == 0
assert viewer.dims.ndim == 2
np.testing.assert_equal(viewer.layers[1].data.shape, (10, 15))
np.testing.assert_equal(viewer.layers[1].scale, (1, 1))
np.testing.assert_equal(viewer.layers[1].translate, (0, 0))
def test_new_labels():
"""Test adding new labels image."""
# Add labels to empty viewer
viewer = ViewerModel()
viewer._new_labels()
assert len(viewer.layers) == 1
assert np.max(viewer.layers[0].data) == 0
# Add labels with image already present
viewer = ViewerModel()
np.random.seed(0)
data = np.random.random((10, 15))
viewer.add_image(data)
viewer._new_labels()
assert len(viewer.layers) == 2
assert np.max(viewer.layers[1].data) == 0
