def test_support_for_other_geometry_column_names(self): df = pd.DataFrame([ { "xxx": Point(0, 0), "t": datetime(2018, 1, 1, 12, 0, 0) }, { "xxx": Point(6, 0), "t": datetime(2018, 1, 1, 12, 6, 0) }, { "xxx": Point(6, 6), "t": datetime(2018, 1, 1, 12, 10, 0) }, ]).set_index("t") geo_df = GeoDataFrame(df, geometry="xxx", crs=CRS_METRIC) traj = Trajectory(geo_df, 1) traj.add_speed() traj.add_direction() traj.get_bbox() traj.get_length() traj.get_linestring_between(datetime(2018, 1, 1, 12, 0, 0), datetime(2018, 1, 1, 12, 6, 0)) traj.get_start_location() traj.hvplot() traj.size() traj.to_line_gdf() traj.to_linestringm_wkt() traj.to_linestring() traj.to_point_gdf() traj.to_traj_gdf()
def test_to_line_gdf(self): df = pd.DataFrame([ { "geometry": Point(0, 0), "t": datetime(2018, 1, 1, 12, 0, 0) }, { "geometry": Point(6, 0), "t": datetime(2018, 1, 1, 12, 6, 0) }, { "geometry": Point(6, 6), "t": datetime(2018, 1, 1, 12, 10, 0) }, ]).set_index("t") geo_df = GeoDataFrame(df, crs=CRS_METRIC) traj = Trajectory(geo_df, 1) line_gdf = traj.to_line_gdf() df2 = pd.DataFrame([ { "t": datetime(2018, 1, 1, 12, 6, 0), "prev_t": datetime(2018, 1, 1, 12, 0, 0), "geometry": LineString([(0, 0), (6, 0)]), }, { "t": datetime(2018, 1, 1, 12, 10, 0), "prev_t": datetime(2018, 1, 1, 12, 6, 0), "geometry": LineString([(6, 0), (6, 6)]), }, ]) expected_line_gdf = GeoDataFrame(df2, crs=CRS_METRIC) assert_frame_equal(line_gdf, expected_line_gdf)