|
4 | 4 | import autoarray as aa |
5 | 5 |
|
6 | 6 |
|
7 | | -def test__in_grid_1d__out_ndarray_2d(): |
8 | | - mask_1d = aa.Mask1D(mask=[True, False, False, True], pixel_scales=(1.0,)) |
9 | | - |
10 | | - grid_1d = aa.Grid1D.from_mask(mask=mask_1d) |
11 | | - |
12 | | - obj = aa.m.MockGrid2DLikeObj() |
13 | | - |
14 | | - ndarray_2d = obj.ndarray_2d_from(grid=grid_1d) |
15 | | - |
16 | | - assert isinstance(ndarray_2d, aa.Grid2D) |
17 | | - assert ndarray_2d.native == pytest.approx( |
18 | | - np.array([[[0.0, 0.0], [0.0, -1.0], [0.0, 1.0], [0.0, 0.0]]]), abs=1.0e-4 |
19 | | - ) |
20 | | - |
21 | | - |
22 | | -def test__in_dgrid_1d__out_ndarray_2d_list(): |
23 | | - mask = aa.Mask1D(mask=[True, False, False, True], pixel_scales=(1.0,)) |
24 | | - |
25 | | - grid_1d = aa.Grid1D.from_mask(mask=mask) |
26 | | - |
27 | | - obj = aa.m.MockGrid2DLikeObj() |
28 | | - |
29 | | - ndarray_2d_list = obj.ndarray_2d_list_from(grid=grid_1d) |
30 | | - |
31 | | - assert isinstance(ndarray_2d_list[0], aa.Grid2D) |
32 | | - assert ndarray_2d_list[0].native == pytest.approx( |
33 | | - np.array([[[0.0, 0.0], [0.0, -0.5], [0.0, 0.5], [0.0, 0.0]]]), abs=1.0e-4 |
34 | | - ) |
35 | | - |
36 | | - assert isinstance(ndarray_2d_list[1], aa.Grid2D) |
37 | | - assert ndarray_2d_list[1].native == pytest.approx( |
38 | | - np.array([[[0.0, 0.0], [0.0, -1.0], [0.0, 1.0], [0.0, 0.0]]]), abs=1.0e-4 |
39 | | - ) |
40 | | - |
41 | | - |
42 | 7 | def test__in_grid_2d__out_ndarray_2d(): |
43 | 8 | mask = aa.Mask2D( |
44 | 9 | mask=[ |
|
0 commit comments