Add submap vs slice test that is failing

sunpy · Apr 25, 2024 · afc8c52 · afc8c52
1 parent 98e0b0a
commit afc8c52
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/sunpy/map/tests/test_mapbase.py b/sunpy/map/tests/test_mapbase.py
@@ -27,6 +27,7 @@
 import sunpy.coordinates
 import sunpy.map
 import sunpy.sun
+from ndcube.tests.helpers import assert_cubes_equal
 from sunpy.coordinates import HeliographicCarrington, HeliographicStonyhurst, sun
 from sunpy.data.test import get_dummy_map_from_header, get_test_filepath
 from sunpy.image.transform import _rotation_registry
@@ -1859,3 +1860,16 @@ def test_set_wcs_modifies_crpix(aia171_test_map, aslice, dims):
                                                               aia171_test_map.meta["CRPIX2"])
 
     assert np.allclose(sliced_ref_coord, ori_ref_coord)
+
+def test_slicing_map_submap(aia171_test_map):
+    bl_coord = SkyCoord(-1000 * u.arcsec, 0 * u.arcsec, frame=aia171_test_map.coordinate_frame)
+    bottom_left = aia171_test_map.wcs.world_to_pixel(bl_coord)
+
+    tp_coord = SkyCoord(0 * u.arcsec, 1000 * u.arcsec, frame=aia171_test_map.coordinate_frame)
+    top_right = aia171_test_map.wcs.world_to_pixel(tp_coord)
+
+    submap = aia171_test_map.submap(bottom_left*u.pix, top_right=top_right*u.pix)
+    sliced_map = aia171_test_map[int(bottom_left[1]):int(top_right[1]) + 1,
+                                 int(bottom_left[0]):int(top_right[0]) + 1]
+
+    assert_cubes_equal(submap, sliced_map, check_data=True)