fix tests to support RGB or RGBA