Fix expected results of get_colour tests

The image is 4 high, we should expect the result to be also.