def test_2_3quarter():
value = 0.75
params = (two_colors, value)
expected_answer = np.array([63.75, 63.75, 63.75])
close_enough(expected_answer, get_gradient(*params), atol=1)
def test_3_3quarter():
value = 0.75
params = (three_colors, value)
expected_answer = np.array([127.5, 0, 0])
close_enough(expected_answer, get_gradient(*params), atol=1)
def test_2_half():
value = 0.5
params = (two_colors, value)
expected_answer = np.array([127.5, 127.5, 127.5])
close_enough(expected_answer, get_gradient(*params), atol=1)
def test_3_half():
value = 0.5
params = (three_colors, value)
expected_answer = np.array([0, 0, 0])
close_enough(expected_answer, get_gradient(*params), atol=1)