def test_azimuth_east(self):
result = azimuth(Point(0, 0), Point(1, 0))
expected_result = 90
self.assertEqual(expected_result, result)
result = azimuth(Point(0, 0), Point(100, 0))
expected_result = 90
self.assertEqual(expected_result, result)
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_azimuth_southeast(self):
assert azimuth(Point(0, 0), Point(1, -1)) == 135
def test_azimuth_northeast(self):
assert azimuth(Point(0, 0), Point(1, 1)) == 45
def test_azimuth_south(self):
assert azimuth(Point(0, 0), Point(0, -1)) == 180
def test_azimuth_north(self):
assert azimuth(Point(0, 0), Point(0, 1)) == 0
def test_azimuth_west(self):
assert azimuth(Point(0, 0), Point(-10, 0)) == 270
def test_azimuth_east(self):
assert azimuth(Point(0, 0), Point(1, 0)) == 90
assert azimuth(Point(0, 0), Point(100, 0)) == 90
def test_azimuth_northwest(self):
result = azimuth(Point(100, 100), Point(99, 101))
expected_result = 315
self.assertEqual(expected_result, result)
def test_azimuth_southwest(self):
result = azimuth(Point(0, 0), Point(-1, -1))
expected_result = 225
self.assertEqual(expected_result, result)
def test_azimuth_northeast(self):
result = azimuth(Point(0, 0), Point(1, 1))
expected_result = 45
self.assertEqual(expected_result, result)
def test_azimuth_south(self):
result = azimuth(Point(0, 0), Point(0, -1))
expected_result = 180
self.assertEqual(expected_result, result)
def test_azimuth_west(self):
result = azimuth(Point(0, 0), Point(-10, 0))
expected_result = 270
self.assertEqual(expected_result, result)
def test_azimuth_southwest(self):
assert azimuth(Point(0, 0), Point(-1, -1)) == 225
def test_azimuth_northwest(self):
assert azimuth(Point(100, 100), Point(99, 101)) == 315
def test_azimuth_throws_type_error(self):
with pytest.raises(TypeError):
azimuth((0, 0), (0, 1))
