def test_setrgba(self):
color = Color(r=1, g=2, b=3, a=1)
color.set_rgba(r=1, g=1, b=1, a=5)
assert color.r == 1
assert color.g == 1
assert color.b == 1
assert color.a == 1.0
def test_setters(self):
color = Color(r=1, g=2, b=3, a=0.1)
color.r = 2
color.g = 3
color.b = 4
color.a = 0.2
assert color.r == 2
assert color.g == 3
assert color.b == 4
assert color.a == 0.2
def get_complement(color: Color, num=1) -> List[Color]:
HSL_list = RGB_to_HSL(color.RGB)
HSL_list_comp = H_rotate(HSL_list, num)
RGB_list_comp = [HSL_to_RGB(hsl) for hsl in HSL_list_comp]
return [Color(r=rgb.R, g=rgb.G, b=rgb.B) for rgb in RGB_list_comp]
def test_mult(self):
color_1 = Color(r=1, g=2, b=3, a=1)
assert color_1 * 2 == Color(r=2, g=4, b=6, a=2)
def test_sub(self):
color_1 = Color(r=1, g=2, b=3, a=100)
color_2 = Color(r=1, g=2, b=3, a=1)
color_3 = color_1 - color_2
assert color_1 - 1 == Color(r=0, g=1, b=2, a=0.0)
assert color_1 - color_2 == Color(r=0, g=0, b=0, a=0.0)
def test_getters(self):
color = Color(r=1, g=2, b=3, a=0.1)
assert color.r == 1
assert color.g == 2
assert color.b == 3
assert color.a == 0.1
def test_add(self):
color_1 = Color(r=1, g=2, b=3, a=1)
color_2 = Color(r=1, g=2, b=3, a=1)
color_3 = color_1 + color_2
assert color_1 + 1 == Color(r=2, g=3, b=4, a=2)
assert color_1 + color_2 == Color(r=2, g=4, b=6, a=2)
def test_eq(self):
color_1 = Color(r=1, g=2, b=3, a=1)
color_2 = Color(r=1, g=2, b=3, a=1)
assert color_1 == color_2
def test_setter_clamp(self):
color = Color(r=266, g=-2, b=333, a=1.2)
assert color.r == 255
assert color.g == 0
assert color.b == 255
assert color.a == 1.0
def invert(color: Color):
R = color.r * -1 + 255
G = color.g * -1 + 255
B = color.b * -1 + 255
return Color(R, G, B)