Skip to content

Commit 6faddd6

Browse files
Jammy2211Jammy2211
authored andcommitted
fix test
1 parent b12b2b9 commit 6faddd6

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

test_autoarray/inversion/inversion/test_mapper_valued.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -70,7 +70,7 @@ def test__interpolated_array_from():
7070

7171

7272
def test__interpolated_array_from__with_pixel_mask():
73-
values = np.array([0.0, 1.0, 1.0, 1.0])
73+
values = jnp.array([0.0, 1.0, 1.0, 1.0])
7474

7575
mapper = aa.m.MockMapper(parameters=4, interpolated_array=values)
7676

@@ -84,7 +84,7 @@ def test__interpolated_array_from__with_pixel_mask():
8484
shape_native=(3, 3), extent=(-0.2, 0.2, -0.3, 0.3)
8585
)
8686

87-
assert (values == np.array([0.0, 1.0, 1.0, 0.0])).all()
87+
assert values == pytest.approx(np.array([0.0, 1.0, 1.0, 1.0]), 1.0e-4)
8888

8989

9090
def test__magnification_via_mesh_from():

0 commit comments

Comments
 (0)