Copied tests to _colors. Moved 'return', pypy

pypy compatibility hack in tests.