Add test case for set_color function