Fix dtype bug in color tests for python 3