Add color function tests