Skip to content

Commit aedb3b2

Browse files
committed
Add numpy conversion for now
1 parent ebf9127 commit aedb3b2

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

ccdproc/tests/test_ccdproc.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1094,7 +1094,9 @@ def test_wcs_project_onto_scale_wcs():
10941094
# TODO: change back to .mask when CCDData is array-api compliant
10951095
ccd_data._mask = xp.zeros_like(ccd_data.data)
10961096
# ...except the center pixel, which is one.
1097-
ccd_data.mask[int(ccd_data.wcs.wcs.crpix[0]), int(ccd_data.wcs.wcs.crpix[1])] = 1
1097+
ccd_data._mask = xpx.at(ccd_data._mask)[
1098+
int(ccd_data.wcs.wcs.crpix[0]), int(ccd_data.wcs.wcs.crpix[1])
1099+
].set(1)
10981100

10991101
target_wcs = wcs_for_testing(ccd_data.shape)
11001102
target_wcs.wcs.cdelt /= 2
@@ -1105,6 +1107,7 @@ def test_wcs_project_onto_scale_wcs():
11051107

11061108
# TODO: rm hack for numpy-specific check in astropy.wcs
11071109
ccd_data.data = np_array(ccd_data.data)
1110+
ccd_data._mask = np_array(ccd_data._mask)
11081111
# Explicitly set the interpolation method so we know what to
11091112
# expect for the mass.
11101113
new_ccd = wcs_project(

0 commit comments

Comments
 (0)