Simplify test
Use __future__ division instead of floats everywhere