Files
PythonRobotics/tests/test_grid_map_lib.py
Atsushi Sakai 0fc769421f Adding NDT mapping script and doc (#867)
* Adding ndt mapping script and doc

* Adding ndt mapping script and doc

* Adding ndt mapping script and doc

* Adding ndt mapping script and doc

* Adding ndt mapping script and doc

* Adding ndt mapping script and doc

* Adding ndt mapping script and doc

* Adding ndt mapping script and doc
2023-07-06 23:12:43 +09:00

41 lines
1.3 KiB
Python

from Mapping.grid_map_lib.grid_map_lib import GridMap
import conftest
import numpy as np
def test_position_set():
grid_map = GridMap(100, 120, 0.5, 10.0, -0.5)
grid_map.set_value_from_xy_pos(10.1, -1.1, 1.0)
grid_map.set_value_from_xy_pos(10.1, -0.1, 1.0)
grid_map.set_value_from_xy_pos(10.1, 1.1, 1.0)
grid_map.set_value_from_xy_pos(11.1, 0.1, 1.0)
grid_map.set_value_from_xy_pos(10.1, 0.1, 1.0)
grid_map.set_value_from_xy_pos(9.1, 0.1, 1.0)
def test_polygon_set():
ox = [0.0, 4.35, 20.0, 50.0, 100.0, 130.0, 40.0]
oy = [0.0, -4.15, -20.0, 0.0, 30.0, 60.0, 80.0]
grid_map = GridMap(600, 290, 0.7, 60.0, 30.5)
grid_map.set_value_from_polygon(ox, oy, 1.0, inside=False)
grid_map.set_value_from_polygon(np.array(ox), np.array(oy),
1.0, inside=False)
def test_xy_and_grid_index_conversion():
grid_map = GridMap(100, 120, 0.5, 10.0, -0.5)
for x_ind in range(grid_map.width):
for y_ind in range(grid_map.height):
grid_ind = grid_map.calc_grid_index_from_xy_index(x_ind, y_ind)
x_ind_2, y_ind_2 = grid_map.calc_xy_index_from_grid_index(grid_ind)
assert x_ind == x_ind_2
assert y_ind == y_ind_2
if __name__ == '__main__':
conftest.run_this_test(__file__)