def test_sp_mean_cupy(): sp_mean = select('sp_mean', backend='cupy') im_test = read_resize(ASSETS('beach24.png'), (224,224))[...,0] segments = np.load(ASSETS('beach_sp.npy')) y = sp_mean(im_test, segments) y_true = np.load(ASSETS('beach_sp_mean.npy')) npt.assert_allclose(y, y_true, rtol=1e-03)
def test_read_resize_shape(): im_test = read_resize(ASSETS('beach24.png'), (224, 224)) assert im_test.shape == (224, 224, 3)
def test_read_resize_range(): im_test = read_resize(ASSETS('beach24.png'), (224, 224)) assert 178 < im_test.mean() < 180
def test_sp_mean(): im_test = read_resize(ASSETS('beach24.png'), (224, 224))[..., 0] segments = np.load(ASSETS('beach_sp.npy')) y = sp_mean(im_test, segments) y_true = np.load(ASSETS('beach_sp_mean.npy')) npt.assert_allclose(y, y_true, rtol=0.005)
def test_read_resize_dtype(): im_test = read_resize(ASSETS('beach24.png'), (224, 224)) assert im_test.dtype == np.dtype('u1')