I think `assert_allclose` needs to be imported from `astropy.tests.compat` up top.