def test_3d(self):
dim_names = ["y", "x", "z"]
waypoints = [
{
dim_names[0]: 0,
dim_names[1]: 0,
dim_names[2]: 2
},
{
dim_names[0]: 1,
dim_names[1]: 2,
dim_names[2]: 10
},
]
sample_count = 5
trajectory = Trajectory(waypoints, sample_count=sample_count)
result = trajectory._get_interp_points()
self.assertEqual(len(result), len(waypoints[0]))
self.assertEqual(len(result[0][1]), sample_count)
self.assertEqual(len(result[1][1]), sample_count)
self.assertEqual(len(result[2][1]), sample_count)
self.assertIn(result[0][0], dim_names)
self.assertIn(result[1][0], dim_names)
self.assertIn(result[2][0], dim_names)
def test_basic(self):
dim_names = "lat"
waypoints = [{dim_names: 0}, {dim_names: 1}]
sample_count = 5
trajectory = Trajectory(waypoints, sample_count=sample_count)
result = trajectory._get_interp_points()
expected_points = list(np.linspace(0, 1, sample_count))
self.assertEqual(len(result), len(waypoints[0]))
self.assertEqual(len(result[0][1]), sample_count)
self.assertEqual(result[0][1], expected_points)
self.assertEqual(result[0][0], dim_names)
def test_basic(self):
dim_names = 'lat'
waypoints = [{dim_names: 0}, {dim_names: 1}]
sample_count = 5
trajectory = Trajectory(waypoints, sample_count=sample_count)
result = trajectory._get_interp_points()
expected_points = list(np.linspace(0, 1, sample_count))
self.assertEqual(len(result), len(waypoints[0]))
self.assertEqual(len(result[0][1]), sample_count)
self.assertEqual(result[0][1], expected_points)
self.assertEqual(result[0][0], dim_names)
def test_2d(self):
dim_names = ['lat', 'lon']
waypoints = [{dim_names[0]: 0, dim_names[1]: 0},
{dim_names[0]: 1, dim_names[1]: 2}]
sample_count = 5
trajectory = Trajectory(waypoints, sample_count=sample_count)
result = trajectory._get_interp_points()
self.assertEqual(len(result), len(waypoints[0]))
self.assertEqual(len(result[0][1]), sample_count)
self.assertEqual(len(result[1][1]), sample_count)
self.assertIn(result[0][0], dim_names)
self.assertIn(result[1][0], dim_names)
def test_2d(self):
dim_names = ['lat', 'lon']
waypoints = [{dim_names[0]: 0, dim_names[1]: 0},
{dim_names[0]: 1, dim_names[1]: 2}]
sample_count = 5
trajectory = Trajectory(waypoints, sample_count=sample_count)
result = trajectory._get_interp_points()
self.assertEqual(len(result), len(waypoints[0]))
self.assertEqual(len(result[0][1]), sample_count)
self.assertEqual(len(result[1][1]), sample_count)
self.assertIn(result[0][0], dim_names)
self.assertIn(result[1][0], dim_names)