33import autoarray as aa
44
55
6- def test__in_grid_1d__out_ndarray_1d ():
7- grid_1d = aa .Grid1D .no_mask (values = [1.0 , 2.0 , 3.0 ], pixel_scales = 1.0 )
8-
9- obj = aa .m .MockGrid1DLikeObj ()
10-
11- ndarray_1d = obj .ndarray_1d_from (grid = grid_1d )
12-
13- assert isinstance (ndarray_1d , aa .Array1D )
14- assert (ndarray_1d .native == np .array ([1.0 , 1.0 , 1.0 ])).all ()
15- assert ndarray_1d .pixel_scales == (1.0 ,)
16-
17- obj = aa .m .MockGrid1DLikeObj (centre = (1.0 , 0.0 ), angle = 45.0 )
18-
19- ndarray_1d = obj .ndarray_1d_from (grid = grid_1d )
20-
21- assert isinstance (ndarray_1d , aa .Array1D )
22- assert (ndarray_1d .native == np .array ([1.0 , 1.0 , 1.0 ])).all ()
23- assert ndarray_1d .pixel_scales == (1.0 ,)
24-
25- mask_1d = aa .Mask1D (mask = [True , False , False , True ], pixel_scales = (1.0 ,))
26-
27- grid_1d = aa .Grid1D .from_mask (mask = mask_1d )
28-
29- obj = aa .m .MockGrid2DLikeObj ()
30-
31- ndarray_1d = obj .ndarray_1d_from (grid = grid_1d )
32-
33- assert isinstance (ndarray_1d , aa .Array1D )
34- assert (ndarray_1d .native == np .array ([0.0 , 1.0 , 1.0 , 0.0 ])).all ()
35-
36-
376def test__in_grid_1d__out_ndarray_1d_list ():
387 mask = aa .Mask1D (mask = [True , False , False , True ], pixel_scales = (1.0 ,))
398
@@ -50,55 +19,6 @@ def test__in_grid_1d__out_ndarray_1d_list():
5019 assert (ndarray_1d_list [1 ].native == np .array ([[0.0 , 2.0 , 2.0 , 0.0 ]])).all ()
5120
5221
53- def test__in_grid_2d__out_ndarray_1d ():
54- grid_2d = aa .Grid2D .uniform (shape_native = (4 , 4 ), pixel_scales = 1.0 )
55-
56- obj = aa .m .MockGrid1DLikeObj ()
57-
58- ndarray_1d = obj .ndarray_1d_from (grid = grid_2d )
59-
60- assert isinstance (ndarray_1d , aa .Array1D )
61- assert (ndarray_1d .native == np .array ([1.0 ])).all ()
62- assert ndarray_1d .pixel_scales == (1.0 ,)
63-
64- obj = aa .m .MockGrid1DLikeObj (centre = (1.0 , 0.0 ))
65-
66- ndarray_1d = obj .ndarray_1d_from (grid = grid_2d )
67-
68- assert isinstance (ndarray_1d , aa .Array1D )
69- assert (ndarray_1d .native == np .array ([1.0 , 1.0 , 1.0 , 1.0 ])).all ()
70- assert ndarray_1d .pixel_scales == (1.0 ,)
71-
72- mask = aa .Mask2D (
73- mask = [
74- [True , True , True , True ],
75- [True , False , False , True ],
76- [True , False , False , True ],
77- [True , True , True , True ],
78- ],
79- pixel_scales = (1.0 , 1.0 ),
80- )
81-
82- grid_2d = aa .Grid2D .from_mask (mask = mask )
83-
84- obj = aa .m .MockGrid2DLikeObj ()
85-
86- ndarray_1d = obj .ndarray_1d_from (grid = grid_2d )
87-
88- assert isinstance (ndarray_1d , aa .Array2D )
89- assert (
90- ndarray_1d .native
91- == np .array (
92- [
93- [0.0 , 0.0 , 0.0 , 0.0 ],
94- [0.0 , 1.0 , 1.0 , 0.0 ],
95- [0.0 , 1.0 , 1.0 , 0.0 ],
96- [0.0 , 0.0 , 0.0 , 0.0 ],
97- ]
98- )
99- ).all ()
100-
101-
10222def test__in_grid_2d__out_ndarray_1d_list ():
10323 mask = aa .Mask2D (
10424 mask = [
0 commit comments