def test__3x3_grid__buffer_aligns_two_grids(self): galaxy = ag.Galaxy(redshift=0.5, light=ag.lp.EllSersic(intensity=1.0)) grid_without_buffer = np.array([[-1.48, -1.48], [1.48, 1.48]]) plane_image = plane_image_of_galaxies_from( shape=(3, 3), grid=grid_without_buffer, galaxies=[galaxy], buffer=0.02 ) mask = ag.Mask2D.manual( mask=np.full(shape=(3, 3), fill_value=False), pixel_scales=1.0, sub_size=1 ) grid = ag.Grid2D.manual_mask( grid=np.array( [ [-1.0, -1.0], [-1.0, 0.0], [-1.0, 1.0], [0.0, -1.0], [0.0, 0.0], [0.0, 1.0], [1.0, -1.0], [1.0, 0.0], [1.0, 1.0], ] ), mask=mask, ) plane_image_galaxy = galaxy.image_2d_from_grid(grid=grid) assert (plane_image.array == plane_image_galaxy).all()
def test__2x3_grid__shape_change_correct_and_coordinates_shift(self): galaxy = ag.Galaxy(redshift=0.5, light=ag.lp.EllSersic(intensity=1.0)) grid = np.array([[-1.5, -1.5], [1.5, 1.5]]) plane_image = plane_image_of_galaxies_from( shape=(2, 3), grid=grid, galaxies=[galaxy], buffer=0.0 ) mask = ag.Mask2D.manual( mask=np.full(shape=(2, 3), fill_value=False), pixel_scales=1.0, sub_size=1 ) grid = ag.Grid2D.manual_mask( grid=np.array( [ [-0.75, -1.0], [-0.75, 0.0], [-0.75, 1.0], [0.75, -1.0], [0.75, 0.0], [0.75, 1.0], ] ), mask=mask, ) plane_image_galaxy = galaxy.image_2d_from_grid(grid=grid) assert (plane_image.array == plane_image_galaxy).all()
def test__3x3_grid__extracts_max_min_coordinates__creates_grid_including_half_pixel_offset_from_edge( self, ): galaxy = ag.Galaxy(redshift=0.5, light=ag.lp.EllSersic(intensity=1.0)) grid = np.array([[-1.5, -1.5], [1.5, 1.5]]) plane_image = plane_image_of_galaxies_from( shape=(3, 3), grid=grid, galaxies=[galaxy], buffer=0.0 ) mask = ag.Mask2D.manual( mask=np.full(shape=(3, 3), fill_value=False), pixel_scales=1.0, sub_size=1 ) grid = ag.Grid2D.manual_mask( grid=np.array( [ [-1.0, -1.0], [-1.0, 0.0], [-1.0, 1.0], [0.0, -1.0], [0.0, 0.0], [0.0, 1.0], [1.0, -1.0], [1.0, 0.0], [1.0, 1.0], ] ), mask=mask, ) plane_image_galaxy = galaxy.image_2d_from_grid(grid) assert (plane_image.array == plane_image_galaxy).all()
def plane_image_2d_from(self, grid: aa.type.Grid2DLike) -> "PlaneImage": return plane_util.plane_image_of_galaxies_from( shape=grid.mask.shape, grid=grid.mask.unmasked_grid_sub_1, galaxies=self.galaxies, )
def plane_image_2d_from_grid(self, grid): return plane_util.plane_image_of_galaxies_from( shape=grid.mask.shape, grid=grid.mask.unmasked_grid_sub_1, galaxies=self.galaxies, )