alihan/textual
1
0
Fork 0
mirror of https://github.com/Textualize/textual.git synced 2025-10-17 02:38:12 +03:00
Code Issues Packages Projects Releases Wiki Activity

tests

Browse Source
This commit is contained in:
Will McGugan
2023-02-17 12:36:35 +00:00
parent edf00a7d0b
commit c65e52be53
2 changed files with 24 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/textual/_spatial_map.py
View File

@@ -68,7 +68,7 @@ class SpatialMap(Generic[ValueType]):
def get_values_in_region(self, region: Region) -> list[ValueType]:
"""Get a set of values that are under a given region.
Note that this may return some false positives.
Note that this may return false positives.
Args:
region: A region.

23
tests/test_spatial_map.py
View File

@@ -13,6 +13,12 @@ from textual.geometry import Region
(0, 0),
],
),
(
Region(10, 10, 10, 10),
[
(1, 1),
],
),
(
Region(0, 0, 11, 11),
[(0, 0), (0, 1), (1, 0), (1, 1)],
@@ -31,3 +37,20 @@ def test_region_to_grid(region, grid):
spatial_map = SpatialMap(10, 10)
assert list(spatial_map._region_to_grid(region)) == grid
def test_get_values_in_region() -> None:
spatial_map: SpatialMap[str] = SpatialMap(20, 10)
spatial_map.insert(
[
(Region(10, 5, 5, 5), False, "foo"),
(Region(5, 20, 5, 5), False, "bar"),
]
)
assert spatial_map.get_values_in_region(Region(0, 0, 10, 5)) == ["foo"]
assert spatial_map.get_values_in_region(Region(0, 1, 10, 5)) == ["foo"]
assert spatial_map.get_values_in_region(Region(0, 10, 10, 5)) == []
assert spatial_map.get_values_in_region(Region(0, 20, 10, 5)) == ["bar"]
assert spatial_map.get_values_in_region(Region(5, 5, 50, 50)) == ["foo", "bar"]