def test_genericmap_mask(aia171_test_map, tmpdir): mask = np.zeros_like(aia171_test_map.data) mask[10, 10] = 1 aia171_test_map.mask = mask aia171_test_map._unit = u.m tree = {'smap': aia171_test_map} assert_roundtrip_tree(tree, tmpdir, extensions=SunpyExtension())
def test_genericmap_basic(aia171_test_map, tmpdir): tree = {'smap': aia171_test_map} assert_roundtrip_tree(tree, tmpdir, extensions=SunpyExtension())