Added `tests` to `TYPES`. Fixed test to run in python2.7