def test_add_heading_latlon(self):
data = [{'geometry':Point(0,0), 't':datetime(2018,1,1,12,0,0)},
{'geometry':Point(10,10), 't':datetime(2018,1,1,12,10,0)}]
df = pd.DataFrame(data).set_index('t')
geo_df = GeoDataFrame(df, crs={'init': '4326'})
traj = Trajectory(1,geo_df)
traj.add_heading()
result = traj.df['heading'].tolist()
expected_result = [44.561451413257714, 44.561451413257714]
self.assertAlmostEqual(result[0], expected_result[0], 5)
def test_add_heading(self):
data = [{'geometry':Point(0,0), 't':datetime(2018,1,1,12,0,0)},
{'geometry':Point(6,0), 't':datetime(2018,1,1,12,10,0)},
{'geometry':Point(6,-6), 't':datetime(2018,1,1,12,20,0)},
{'geometry':Point(-6,-6), 't':datetime(2018,1,1,12,20,0)}]
df = pd.DataFrame(data).set_index('t')
geo_df = GeoDataFrame(df, crs={'init': '31256'})
traj = Trajectory(1,geo_df)
traj.add_heading()
result = traj.df['heading'].tolist()
expected_result = [90.0, 90.0, 180.0, 270]
self.assertEqual(result, expected_result)
