argmax/argmin need to return ints [some test code was removed]

radio-astro-tools · Oct 18, 2024 · c636655 · c636655
1 parent 0a3cffc
commit c636655
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/spectral_cube/tests/test_spectral_cube.py b/spectral_cube/tests/test_spectral_cube.py
@@ -657,6 +657,14 @@ def test_argmax(self):
         self._check_numpy(self.c.argmax, d, np.nanargmax)
         self.c = self.d = None
 
+    def test_argmax_rays(self):
+        """
+        regression test: argmax must have integer dtype
+        """
+        d = np.where(self.d > 0.5, self.d, -10)
+        result = self.c.argmax(how='ray')
+        assert 'int' in str(result.dtype)
+
     def test_argmin(self):
         d = np.where(self.d > 0.5, self.d, 10)
         self._check_numpy(self.c.argmin, d, np.nanargmin)