Add further tests, thanks @jlaehne

hyperspy · Jun 16, 2022 · c66daa3 · c66daa3
1 parent 5651909
commit c66daa3
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/hyperspy/tests/axes/test_axes_manager.py b/hyperspy/tests/axes/test_axes_manager.py
@@ -20,7 +20,7 @@
 
 import numpy as np
 
-from hyperspy.axes import AxesManager, _serpentine_iter, _flyback_iter, GeneratorLen
+from hyperspy.axes import AxesManager, _serpentine_iter, _flyback_iter, GeneratorLen, BaseDataAxis
 from hyperspy.defaults_parser import preferences
 from hyperspy.signals import BaseSignal, Signal1D, Signal2D
 
@@ -114,6 +114,16 @@ def test_all_uniform(self):
         self.am[-1].convert_to_non_uniform_axis()
         assert self.am.all_uniform == False
 
+    def test_get_axis(self):
+        am = self.am
+        assert am[0] == am["d"]
+        assert am[-1] == am["b"]
+        axis = am[1]
+        assert am[axis] == axis
+        with pytest.raises(ValueError):
+            axis = BaseDataAxis()
+            am[axis]
+
 
 class TestAxesManagerScaleOffset:
     def test_low_high_value(self):