Import pytest from astropy