Changed test to units supported for an older version of astropy (0.32)