FIX: Py3 / pep8 fixes in test_ball_tree / test_kd_tree