def test_changing_labels_dims(): """Test changing Labels data including dimensionality.""" shape_a = (10, 15) shape_b = (20, 12, 6) np.random.seed(0) data_a = np.random.randint(20, size=shape_a) data_b = np.random.randint(20, size=shape_b) layer = Labels(data_a) # Prep indices for swtich to 3D layer._indices = (0, ) + layer._indices layer.data = data_b assert np.all(layer.data == data_b) assert layer.ndim == len(shape_b) assert layer.shape == shape_b assert layer.range == tuple((0, m, 1) for m in shape_b) assert layer._data_view.shape == shape_b[-2:]