def test__from_config_or_via_manual_input(self):
grid_scatter = aplt.GridScatter()
assert grid_scatter.config_dict["marker"] == "x"
assert grid_scatter.config_dict["c"] == "y"
grid_scatter = aplt.GridScatter(marker="x")
assert grid_scatter.config_dict["marker"] == "x"
assert grid_scatter.config_dict["c"] == "y"
grid_scatter = aplt.GridScatter()
grid_scatter.is_for_subplot = True
assert grid_scatter.config_dict["marker"] == "."
assert grid_scatter.config_dict["c"] == "r"
grid_scatter = aplt.GridScatter(c=["r", "b"])
grid_scatter.is_for_subplot = True
assert grid_scatter.config_dict["marker"] == "."
assert grid_scatter.config_dict["c"] == ["r", "b"]
def test__scatter_colored_grid__lists_of_coordinates_or_equivalent_2d_grids__with_color_array(
self, ):
scatter = aplt.GridScatter(s=2, marker="x", c="k")
cmap = plt.get_cmap("jet")
scatter.scatter_grid_colored(
grid=aa.Grid2DIrregular([(1.0, 1.0), (2.0, 2.0), (3.0, 3.0),
(4.0, 4.0), (5.0, 5.0)]),
color_array=np.array([2.0, 2.0, 2.0, 2.0, 2.0]),
cmap=cmap,
)
scatter.scatter_grid_colored(
grid=aa.Grid2D.uniform(shape_native=(3, 2), pixel_scales=1.0),
color_array=np.array([2.0, 2.0, 2.0, 2.0, 2.0, 2.0]),
cmap=cmap,
)
def test__scatter_grid_indexes_1d__input_grid_is_ndarray_and_indexes_are_valid(
self, ):
scatter = aplt.GridScatter(s=2, marker="x", c="k")
scatter.scatter_grid_indexes(
grid=aa.Grid2D.uniform(shape_native=(3, 3), pixel_scales=1.0),
indexes=[0, 1, 2],
)
scatter.scatter_grid_indexes(
grid=aa.Grid2D.uniform(shape_native=(3, 3), pixel_scales=1.0),
indexes=[[0, 1, 2]],
)
scatter.scatter_grid_indexes(
grid=aa.Grid2D.uniform(shape_native=(3, 3), pixel_scales=1.0),
indexes=[[0, 1], [2]],
)
def test__scatter_grid(self):
scatter = aplt.GridScatter(s=2, marker="x", c="k")
scatter.scatter_grid(
grid=aa.Grid2D.uniform(shape_native=(3, 3), pixel_scales=1.0))
def test__scatter_coordinates(self):
scatter = aplt.GridScatter(s=2, marker="x", c="k")
scatter.scatter_grid_list(
grid_list=[aa.Grid2DIrregular([(1.0, 1.0), (2.0, 2.0)])])