def test_saturation():
rgb = _make_array(0.392156, 0.776470, 0.164705)
saturated = _make_array(0.3425, 0.78372, 0.0)
assert np.allclose(saturate_rgb(rgb, 1.1), saturated, atol=0.2)
rgb = _make_array(0.0392, 0.1960, 0.3529)
saturated = _make_array(0.0456, 0.1929, 0.3941)
assert np.allclose(saturate_rgb(rgb, 1.25), saturated, atol=0.2)
def test_bad_array_type():
bad = np.random.random((3, 3, 3)).astype("uint8")
with pytest.raises(ValueError) as exc:
saturate_rgb(bad, 1.1)
assert "dtype mismatch" in str(exc.value)
with pytest.raises(ValueError) as exc:
convert_arr(bad, cs.rgb, cs.lch)
assert "dtype mismatch" in str(exc.value)
def test_bad_array_dims():
bad = np.random.random((3, 3))
with pytest.raises(ValueError) as exc:
saturate_rgb(bad, 1.1)
assert "wrong number of dimensions" in str(exc.value)
with pytest.raises(ValueError) as exc:
convert_arr(bad, cs.rgb, cs.lch)
assert "wrong number of dimensions" in str(exc.value)
def test_bad_array_bands():
bad = np.random.random((2, 3, 3))
with pytest.raises(ValueError) as exc:
saturate_rgb(bad, 1.1)
assert "3 bands" in str(exc.value)
with pytest.raises(ValueError) as exc:
convert_arr(bad, cs.rgb, cs.lch)
assert "3 bands" in str(exc.value)
def test_bad_array_type():
bad = np.random.random((3, 3, 3)).astype('uint8')
with pytest.raises(ValueError) as exc:
saturate_rgb(bad, 1.1)
assert 'dtype mismatch' in str(exc.value)
with pytest.raises(ValueError) as exc:
convert_arr(bad, cs.rgb, cs.lch)
assert 'dtype mismatch' in str(exc.value)
def test_bad_array_dims():
bad = np.random.random((3, 3))
with pytest.raises(ValueError) as exc:
saturate_rgb(bad, 1.1)
assert 'wrong number of dimensions' in str(exc.value)
with pytest.raises(ValueError) as exc:
convert_arr(bad, cs.rgb, cs.lch)
assert 'wrong number of dimensions' in str(exc.value)
def test_bad_array_bands():
bad = np.random.random((2, 3, 3))
with pytest.raises(ValueError) as exc:
saturate_rgb(bad, 1.1)
assert '3 bands' in str(exc.value)
with pytest.raises(ValueError) as exc:
convert_arr(bad, cs.rgb, cs.lch)
assert '3 bands' in str(exc.value)
def test_saturation():
rgb = _make_array(0.392156, 0.776470, 0.164705)
saturated = _make_array(0.3425, 0.78372, 0.0)
assert np.allclose(
saturate_rgb(rgb, 1.1),
saturated,
atol=0.2)
rgb = _make_array(0.0392, 0.1960, 0.3529)
saturated = _make_array(0.0456, 0.1929, 0.3941)
assert np.allclose(
saturate_rgb(rgb, 1.25),
saturated,
atol=0.2)
def test_saturation_bw():
rgb = _make_array(0.392156, 0.776470, 0.164705)
sat = saturate_rgb(rgb, 0.0)
assert _near((sat[0, 0, 0], ), (sat[1, 0, 0], ), tol=0.1)
assert _near((sat[1, 0, 0], ), (sat[2, 0, 0], ), tol=0.1)
def test_saturation_1(pair):
rgb, lch = pair
rgb = _make_array(*rgb)
assert np.allclose(saturate_rgb(rgb, 1.0), rgb, atol=0.2)
def test_saturation_bw():
rgb = _make_array(0.392156, 0.776470, 0.164705)
sat = saturate_rgb(rgb, 0.0)
assert _near((sat[0, 0, 0], ), (sat[1, 0, 0], ), tol=0.1)
assert _near((sat[1, 0, 0], ), (sat[2, 0, 0], ), tol=0.1)
def test_saturation_1(pair):
rgb, lch = pair
rgb = _make_array(*rgb)
assert np.allclose(
saturate_rgb(rgb, 1.0), rgb, atol=0.2)
