Skip to content

Commit 580be1c

Browse files
committed
fix test overlay
1 parent 119866d commit 580be1c

2 files changed

Lines changed: 5 additions & 5 deletions

File tree

test_autoarray/inversion/inversion/imaging/test_imaging.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -11,9 +11,9 @@
1111
directory = path.dirname(path.realpath(__file__))
1212

1313

14-
def test__operated_mapping_matrix_property(psf_7x7, rectangular_mapper_7x7_3x3):
14+
def test__operated_mapping_matrix_property(psf_3x3, rectangular_mapper_7x7_3x3):
1515
inversion = aa.m.MockInversionImaging(
16-
psf=psf_7x7, linear_obj_list=[rectangular_mapper_7x7_3x3]
16+
psf=psf_3x3, linear_obj_list=[rectangular_mapper_7x7_3x3]
1717
)
1818

1919
assert inversion.operated_mapping_matrix_list[0][0, 0] == pytest.approx(1.0, 1e-4)
@@ -42,7 +42,7 @@ def test__operated_mapping_matrix_property(psf_7x7, rectangular_mapper_7x7_3x3):
4242

4343

4444
def test__operated_mapping_matrix_property__with_operated_mapping_matrix_override(
45-
psf_7x7, rectangular_mapper_7x7_3x3
45+
psf_3x3, rectangular_mapper_7x7_3x3
4646
):
4747
psf = aa.m.MockPSF(operated_mapping_matrix=np.ones((2, 2)))
4848

test_autoarray/inversion/pixelization/image_mesh/test_overlay.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -323,13 +323,13 @@ def test__image_plane_mesh_grid_from__simple():
323323

324324
total_pixels = overlay_util.total_pixels_2d_from(
325325
mask_2d=mask.array,
326-
overlaid_centres=overlaid_centres,
326+
overlaid_centres=np.array(overlaid_centres),
327327
)
328328

329329
overlay_for_mask_2d_util = overlay_util.overlay_for_mask_from(
330330
total_pixels=total_pixels,
331331
mask=mask.array,
332-
overlaid_centres=overlaid_centres,
332+
overlaid_centres=np.array(overlaid_centres),
333333
).astype("int")
334334

335335
image_mesh_util = overlay_util.overlay_via_unmasked_overlaid_from(

0 commit comments

Comments
 (0)