def get_direction(self):
"""Return compass bearing as float of Trajectory object."""
pt0 = self.get_start_location()
pt1 = self.get_end_location()
if self.is_latlon():
return calculate_initial_compass_bearing(pt0, pt1)
else:
return azimuth(pt0, pt1)
def _compute_heading(self, row):
pt0 = row['prev_pt']
pt1 = row['geometry']
if type(pt0) != Point:
return 0.0
if pt0 == pt1:
return 0.0
if self.is_latlon():
return calculate_initial_compass_bearing(pt0, pt1)
else:
return azimuth(pt0, pt1)
def test_compass_bearing_south(self):
assert calculate_initial_compass_bearing(Point(0, 0),
Point(0, -10)) == 180
def test_compass_bearing_north(self):
assert calculate_initial_compass_bearing(Point(0, 0), Point(0,
10)) == 0
def test_compass_bearing_west(self):
assert calculate_initial_compass_bearing(Point(0, 0), Point(-10,
0)) == 270
def test_calculate_initial_compass_bearing_throws_type_error(self):
with pytest.raises(TypeError):
calculate_initial_compass_bearing((0, 0), (0, 1))
def test_compass_bearing_south(self):
result = calculate_initial_compass_bearing(Point(0, 0), Point(0, -10))
expected_result = 180
self.assertEqual(expected_result, result)
def test_compass_bearing_west(self):
result = calculate_initial_compass_bearing(Point(0, 0), Point(-10, 0))
expected_result = 270
self.assertEqual(expected_result, result)