def test_operations(self):
col1 = Color(1.0, 2.0, 3.0)
col2 = Color(5.0, 7.0, 9.0)
assert (col1 + col2).is_close(Color(6.0, 9.0, 12.0))
assert (col1 - col2).is_close(Color(-4.0, -5.0, -6.0))
assert (col1 * col2).is_close(Color(5.0, 14.0, 27.0))
prod_col = Color(1.0, 2.0, 3.0) * 2.0
assert prod_col.is_close(Color(2.0, 4.0, 6.0))
def test_create(self):
col = Color(1.0, 2.0, 3.0)
assert col.is_close(Color(1.0, 2.0, 3.0))
def test_close(self):
col = Color(1.0, 2.0, 3.0)
assert not col.is_close(Color(3.0, 4.0, 5.0))
def test_get_set_pixel(self):
img = HdrImage(7, 4)
reference_color = Color(1.0, 2.0, 3.0)
img.set_pixel(3, 2, reference_color)
assert reference_color.is_close(img.get_pixel(3, 2))
