diff --git a/drizzle/tests/test_drizzle.py b/drizzle/tests/test_drizzle.py index 9de9191..1a22861 100644 --- a/drizzle/tests/test_drizzle.py +++ b/drizzle/tests/test_drizzle.py @@ -38,7 +38,10 @@ def centroid(image, size, center): xhi = min(xlo + size, image.shape[1]) yx1 = np.mgrid[ylo:yhi, xlo:xhi, 1:2] - center = (yx1[..., 0] * image[ylo:yhi, xlo:xhi]).sum(axis=(1, 2)) + center = (yx1[..., 0] * image[ylo:yhi, xlo:xhi]).sum( + axis=(1, 2), + dtype=np.float64, + ) if center[2] == 0.0: return None