Fixes to tests_spatial.py