Added a test for add_celestial_axis