Skip to content

Commit 15cd0b5

Browse files
committed
fix some unit tests
1 parent 61a72b8 commit 15cd0b5

7 files changed

Lines changed: 16 additions & 14 deletions

File tree

autoarray/mask/derive/grid_2d.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -112,7 +112,7 @@ def all_false(self) -> Grid2D:
112112
origin=self.mask.origin,
113113
)
114114

115-
return Grid2D(values=grid_slim, mask=self.mask.derive_mask.all_false)
115+
return Grid2D(values=np.array(grid_slim), mask=self.mask.derive_mask.all_false)
116116

117117
@property
118118
def unmasked(self) -> Grid2D:
@@ -158,12 +158,12 @@ def unmasked(self) -> Grid2D:
158158
"""
159159
from autoarray.structures.grids.uniform_2d import Grid2D
160160

161-
grid_1d = grid_2d_util.grid_2d_slim_via_mask_from(
162-
mask_2d=np.array(self.mask),
161+
grid_2d = grid_2d_util.grid_2d_slim_via_mask_from(
162+
mask_2d=self.mask,
163163
pixel_scales=self.mask.pixel_scales,
164164
origin=self.mask.origin,
165165
)
166-
return Grid2D(values=grid_1d, mask=self.mask)
166+
return Grid2D(values=np.array(grid_2d), mask=self.mask)
167167

168168
@property
169169
def edge(self) -> Grid2D:

autoarray/operators/over_sampling/over_sample_util.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
from __future__ import annotations
22
import numpy as np
3-
from typing import TYPE_CHECKING, Union, List, Tuple
3+
from typing import TYPE_CHECKING, Union
44
from typing import List, Tuple
55

66
from autoarray.structures.arrays.uniform_2d import Array2D
@@ -12,7 +12,6 @@
1212
from autoarray.mask.mask_2d import Mask2D
1313

1414
from autoarray import numba_util
15-
from autoarray.mask import mask_2d_util
1615

1716
from autoarray import type as ty
1817

@@ -601,7 +600,7 @@ def over_sample_size_via_radial_bins_from(
601600
radial_grid = grid.distances_to_coordinate_from(coordinate=centre)
602601

603602
sub_size_of_centre = sub_size_radial_bins_from(
604-
radial_grid=np.array(radial_grid),
603+
radial_grid=np.array(radial_grid.array),
605604
sub_size_list=np.array(sub_size_list),
606605
radial_list=np.array(radial_list),
607606
)

autoarray/structures/arrays/uniform_1d.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -88,7 +88,7 @@ def no_mask(
8888
origin=origin,
8989
)
9090

91-
return Array1D(values=values, mask=mask, header=header)
91+
return Array1D(values=np.array(values), mask=mask, header=header)
9292

9393
@classmethod
9494
def full(

autoarray/structures/arrays/uniform_2d.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -692,7 +692,7 @@ def no_mask(
692692
origin=origin,
693693
)
694694

695-
return Array2D(values=values, mask=mask, header=header)
695+
return Array2D(values=np.array(values), mask=mask, header=header)
696696

697697
@classmethod
698698
def full(

autoarray/structures/grids/uniform_1d.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -179,7 +179,7 @@ def no_mask(
179179
origin=origin,
180180
)
181181

182-
return Grid1D(values=values, mask=mask)
182+
return Grid1D(values=np.array(values), mask=mask)
183183

184184
@classmethod
185185
def from_mask(cls, mask: Mask1D) -> "Grid1D":
@@ -196,12 +196,12 @@ def from_mask(cls, mask: Mask1D) -> "Grid1D":
196196
"""
197197

198198
grid_1d = grid_1d_util.grid_1d_slim_via_mask_from(
199-
mask_1d=np.array(mask),
199+
mask_1d=mask.array,
200200
pixel_scales=mask.pixel_scales,
201201
origin=mask.origin,
202202
)
203203

204-
return Grid1D(values=grid_1d, mask=mask)
204+
return Grid1D(values=np.array(grid_1d), mask=mask)
205205

206206
@classmethod
207207
def uniform(

autoarray/structures/grids/uniform_2d.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -247,7 +247,7 @@ def no_mask(
247247
)
248248

249249
return Grid2D(
250-
values=values,
250+
values=np.array(values),
251251
mask=mask,
252252
over_sample_size=over_sample_size,
253253
)
@@ -548,7 +548,7 @@ def from_mask(
548548
"""
549549

550550
grid_2d = grid_2d_util.grid_2d_slim_via_mask_from(
551-
mask_2d=mask,
551+
mask_2d=mask.array,
552552
pixel_scales=mask.pixel_scales,
553553
origin=mask.origin,
554554
)

autoarray/structures/vectors/irregular.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -44,6 +44,9 @@ def __init__(
4444
grid
4545
The irregular grid of (y,x) coordinates where each vector is located.
4646
"""
47+
if type(values) is list:
48+
values = np.asarray(values)
49+
4750
self.grid = Grid2DIrregular(values=grid)
4851

4952
super().__init__(values)

0 commit comments

Comments
 (0)