Skip to content

Commit 7defc3c

Browse files
Jammy2211Jammy2211
authored andcommitted
simplify test_transformer a lot
1 parent 6543589 commit 7defc3c

File tree

1 file changed

+12
-0
lines changed

1 file changed

+12
-0
lines changed

test_autoarray/operators/test_transformer.py

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -153,6 +153,18 @@ def test__nufft__visibilities_from():
153153
assert visibilities_nufft[0] == pytest.approx(25.02317617953263 + 0.0j, 1.0e-7)
154154

155155

156+
def test__nufft__image_from(visibilities_7, uv_wavelengths_7x2, mask_2d_7x7):
157+
158+
transformer = aa.TransformerNUFFT(
159+
uv_wavelengths=uv_wavelengths_7x2,
160+
real_space_mask=mask_2d_7x7,
161+
)
162+
163+
image = transformer.image_from(visibilities=visibilities_7)
164+
165+
assert image[0:3] == pytest.approx([0.00726546, 0.01149121, 0.01421022], 1.0e-4)
166+
167+
156168
def test__nufft__transform_mapping_matrix():
157169
uv_wavelengths = np.array([[0.2, 1.0], [0.5, 1.1], [0.8, 1.2]])
158170

0 commit comments

Comments
 (0)