Fix errors in tranfer of test_celestial.py from TreeCorr