Update test_color.py