Merge pull request #87 from astrofrog/spectral-coordinates-unit

Fix world_axis_units for SpectralCoordinates
glue-viz · Mar 2, 2023 · 478451f · 478451f
2 parents f7b389c + 2125340
commit 478451f
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/glue_astronomy/spectral_coordinates.py b/glue_astronomy/spectral_coordinates.py
@@ -27,6 +27,10 @@ def spectral_axis(self):
         """
         return self._values
 
+    @property
+    def world_axis_units(self):
+        return (self._values.unit.to_string('vounit'),)
+
     def world_to_pixel_values(self, *world):
         """
         Parameters

diff --git a/glue_astronomy/tests/test_spectral_coordinates.py b/glue_astronomy/tests/test_spectral_coordinates.py
@@ -17,6 +17,8 @@ def test_basic():
     assert_allclose(sc.world_to_pixel_values(10), 0)
     assert_allclose(sc.world_to_pixel_values([10, 15, 20]), [0, 0.5, 1.0])
 
+    assert sc.world_axis_units == ('Hz',)
+
 
 def test_invalid_init():