def test_to_traj_gdf(self):
temp_df = self.geo_df.drop(columns=["obj", "val", "val2"])
tc = TrajectoryCollection(temp_df, "id")
traj_gdf = tc.to_traj_gdf()
traj_gdf.to_file("temp.gpkg", layer="trajs", driver="GPKG")
rows = [
{
"traj_id": 1,
"start_t": datetime(2018, 1, 1, 12, 0, 0),
"end_t": datetime(2018, 1, 1, 14, 15, 0),
"geometry": LineString([(0, 0), (6, 0), (6, 6), (9, 9)]),
"length": 12 + sqrt(18),
"direction": 45.0,
},
{
"traj_id": 2,
"start_t": datetime(2018, 1, 1, 12, 0, 0),
"end_t": datetime(2018, 1, 2, 13, 15, 0),
"geometry": LineString([(10, 10), (16, 10), (16, 16),
(190, 10)]),
"length": 12 + sqrt(174 * 174 + 36),
"direction": 90.0,
},
]
df2 = pd.DataFrame(rows)
expected_line_gdf = GeoDataFrame(df2, crs=CRS_METRIC)
assert_frame_equal(traj_gdf, expected_line_gdf)
def test_get_start_locations(self):
collection = TrajectoryCollection(self.geo_df, 'id', obj_id_col='obj')
locs = collection.get_start_locations(columns=['val'])
assert len(locs) == 2
assert locs.iloc[0].geometry in [Point(0, 0), Point(10, 10)]
assert locs.iloc[0].traj_id in [1, 2]
assert locs.iloc[0].obj_id == 'A'
assert locs.iloc[0].val in [9, 10]
assert locs.iloc[1].geometry in [Point(0, 0), Point(10, 10)]
def setup_method(self):
pt_coords = [(0, 0), (6, 0), (6, 6), (0.2, 0.2), (6.2, 0.2),
(6.2, 6.2)]
self.pts = [Point(x, y) for (x, y) in pt_coords]
lst = [
[1, "A", Point(0, 0),
datetime(2018, 1, 1, 12, 0, 0)],
[1, "A", Point(6, 0),
datetime(2018, 1, 1, 12, 6, 0)],
[1, "A", Point(6, 6),
datetime(2018, 1, 1, 14, 10, 0)],
[2, "A", Point(0.2, 0.2),
datetime(2019, 1, 1, 12, 0, 0)],
[2, "A", Point(6.2, 0.2),
datetime(2019, 1, 1, 12, 6, 0)],
[2, "A", Point(6.2, 6.2),
datetime(2019, 1, 1, 14, 10, 0)],
]
df = DataFrame(lst, columns=["id", "obj", "geometry",
"t"]).set_index("t")
self.geo_df = GeoDataFrame(df, crs=CRS_METRIC)
self.collection = TrajectoryCollection(self.geo_df,
"id",
obj_id_col="obj")
self.trajectory_aggregator = TrajectoryCollectionAggregator(
self.collection, 5, 0, min_stop_duration=timedelta(hours=12))
self.expected_clusters = GeoDataFrame(
DataFrame(
[[Point(0.1, 0.1), 2], [Point(6.1, 0.1), 2],
[Point(6.1, 6.1), 2]],
columns=["geometry", "n"],
))
self.expected_flows = GeoDataFrame(
DataFrame(
[
[LineString([Point(0.1, 0.1),
Point(6.1, 0.1)]), 2],
[LineString([Point(6.1, 0.1),
Point(6.1, 6.1)]), 2],
],
columns=["geometry", "weight"],
))
self.geo_df_latlon = GeoDataFrame(df, crs=CRS_LATLON)
self.geo_df_latlon.crs = CRS_LATLON # try to fix
# https://travis-ci.com/github/anitagraser/movingpandas/builds/177149345
self.collection_latlon = TrajectoryCollection(self.geo_df_latlon,
"id",
obj_id_col="obj")
self.trajectory_aggregator_latlon = TrajectoryCollectionAggregator(
self.collection_latlon,
100,
0,
min_stop_duration=timedelta(hours=12))
def setup_method(self):
df = pd.DataFrame([
{'id': 1, 'obj': 'A', 'geometry': Point(0, 0), 't': datetime(2018,1,1,12,0,0), 'val': 9, 'val2': 'a'},
{'id': 1, 'obj': 'A', 'geometry': Point(6, 0), 't': datetime(2018,1,1,12,6,0), 'val': 5, 'val2': 'b'},
{'id': 1, 'obj': 'A', 'geometry': Point(6, 6), 't': datetime(2018,1,1,14,10,0), 'val': 2, 'val2': 'c'}
]).set_index('t')
self.geo_df = GeoDataFrame(df, crs=CRS_METRIC)
self.collection = TrajectoryCollection(self.geo_df, 'id', obj_id_col='obj')
self.trajectory_aggregator = TrajectoryCollectionAggregator(self.collection, 100, 0, min_stop_duration=timedelta(hours=12).seconds)
self.geo_df_latlon = GeoDataFrame(df, crs=CRS_LATLON)
self.geo_df_latlon.crs = CRS_LATLON # try to fix https://travis-ci.com/github/anitagraser/movingpandas/builds/177149345
self.collection_latlon = TrajectoryCollection(self.geo_df_latlon, 'id', obj_id_col='obj')
self.trajectory_aggregator_latlon = TrajectoryCollectionAggregator(self.collection_latlon, 100, 0, min_stop_duration=timedelta(hours=12).seconds)
def setup_method(self):
df = pd.DataFrame([
{'id': 1, 'obj': 'A', 'geometry': Point(0, 0), 't': datetime(2018,1,1,12,0,0), 'val': 9, 'val2': 'a'},
{'id': 1, 'obj': 'A', 'geometry': Point(6, 0), 't': datetime(2018,1,1,12,6,0), 'val': 5, 'val2': 'b'},
{'id': 1, 'obj': 'A', 'geometry': Point(6, 6), 't': datetime(2018,1,1,14,10,0), 'val': 2, 'val2': 'c'},
{'id': 1, 'obj': 'A', 'geometry': Point(9, 9), 't': datetime(2018,1,1,14,15,0), 'val': 4, 'val2': 'd'},
{'id': 2, 'obj': 'A', 'geometry': Point(10, 10), 't': datetime(2018,1,1,12,0,0), 'val': 10, 'val2': 'e'},
{'id': 2, 'obj': 'A', 'geometry': Point(16, 10), 't': datetime(2018,1,1,12,6,0), 'val': 6, 'val2': 'f'},
{'id': 2, 'obj': 'A', 'geometry': Point(16, 16), 't': datetime(2018,1,2,13,10,0), 'val': 7, 'val2': 'g'},
{'id': 2, 'obj': 'A', 'geometry': Point(190, 19), 't': datetime(2018,1,2,13,15,0), 'val': 3, 'val2': 'h'}
]).set_index('t')
self.geo_df = GeoDataFrame(df, crs=CRS_METRIC)
self.collection = TrajectoryCollection(self.geo_df, 'id', obj_id_col='obj')
self.geo_df_latlon = GeoDataFrame(df, crs=CRS_LATLON)
self.collection_latlon = TrajectoryCollection(self.geo_df_latlon, 'id', obj_id_col='obj')
def setup_method(self):
df = pd.DataFrame(
[
[1, "A",
Point(0, 0),
datetime(2018, 1, 1, 12, 0, 0), 3, 10],
[1, "A",
Point(1, 1),
datetime(2018, 1, 1, 12, 1, 0), 1, 14],
[1, "A",
Point(3, 1),
datetime(2018, 1, 1, 12, 2, 0), 3, 11],
[1, "A",
Point(6, 0),
datetime(2018, 1, 1, 12, 6, 0), -10, 20],
[
1, "A",
Point(6, 6),
datetime(2018, 1, 1, 14, 10, 0), 20, 200
],
[1, "A",
Point(9, 9),
datetime(2018, 1, 1, 14, 15, 0), 2, 20],
[
2, "A",
Point(10, 10),
datetime(2018, 1, 1, 12, 0, 0), 30, 20
],
[
2, "A",
Point(16, 10),
datetime(2018, 1, 1, 12, 6, 0), 30, 30
],
[
2, "A",
Point(16, 12),
datetime(2018, 1, 1, 12, 9, 0), 40, 30
],
[
2, "A",
Point(20, 20),
datetime(2018, 1, 1, 12, 20, 0), 1000, 30
],
[
2, "A",
Point(16, 16),
datetime(2018, 1, 2, 13, 10, 0), 20, -30
],
[
2, "A",
Point(190, 19),
datetime(2018, 1, 2, 13, 15, 0), 20, 50
],
],
columns=["id", "obj", "geometry", "t", "val", "val2"],
).set_index("t")
self.geo_df = GeoDataFrame(df, crs=CRS_METRIC)
self.collection = TrajectoryCollection(self.geo_df,
"id",
obj_id_col="obj")
def test_stop_detector_collection(self):
traj1 = make_traj([
Node(0, 0),
Node(0, 1, second=1),
Node(0, 2, second=2),
Node(0, 1, second=3),
Node(0, 22, second=4),
Node(0, 30, second=8),
Node(0, 40, second=10),
Node(1, 50, second=15)
],
id=1)
traj2 = make_traj([
Node(0, -100),
Node(0, -10, second=1),
Node(0, 2, second=2),
Node(0, 1, second=3),
Node(0, 22, second=4),
Node(0, 30, second=8),
Node(0, 31, second=10),
Node(1, 32, second=15)
],
id=2)
collection = TrajectoryCollection([traj1, traj2])
detector = TrajectoryStopDetector(collection)
stop_times = detector.get_stop_time_ranges(
max_diameter=3, min_duration=timedelta(seconds=2))
stop_segments = detector.get_stop_segments(
max_diameter=3, min_duration=timedelta(seconds=2))
assert len(stop_times) == 2
assert len(stop_segments) == 2
def setup_method(self):
df = pd.DataFrame(
[
[1, "A", Point(0, 0),
datetime(2018, 1, 1, 12, 0, 0)],
[1, "A", Point(6, 0),
datetime(2018, 1, 1, 12, 6, 0)],
[1, "A", Point(6, 6),
datetime(2018, 1, 1, 14, 10, 0)],
[1, "A", Point(9, 9),
datetime(2018, 1, 1, 14, 15, 0)],
[2, "A", Point(10, 10),
datetime(2018, 1, 1, 12, 0, 0)],
[2, "A", Point(16, 10),
datetime(2018, 1, 1, 12, 6, 0)],
[2, "A",
Point(16, 16),
datetime(2018, 1, 2, 13, 10, 0)],
[2, "A",
Point(190, 19),
datetime(2018, 1, 2, 13, 15, 0)],
],
columns=["id", "obj", "geometry", "t"],
).set_index("t")
self.df = GeoDataFrame(df, crs=CRS_METRIC)
self.collection = TrajectoryCollection(self.df, "id", obj_id_col="obj")
def test_traj_with_less_than_two_points(self):
df = pd.DataFrame([
{'id': 1, 'obj': 'A', 'geometry': Point(0, 0), 't': datetime(2018,1,1,12,0,0), 'val': 9, 'val2': 'a'}
]).set_index('t')
geo_df = GeoDataFrame(df, crs=CRS_METRIC)
tc = TrajectoryCollection(geo_df, 'id', obj_id_col='obj')
assert len(tc) == 0
def setup_method(self):
df = pd.DataFrame(
[
[1, "A",
Point(0, 0),
datetime(2018, 1, 1, 12, 0, 0), 9, "a"],
[1, "A",
Point(6, 0),
datetime(2018, 1, 1, 12, 6, 0), 5, "b"],
[1, "A",
Point(6, 6),
datetime(2018, 1, 1, 14, 10, 0), 2, "c"],
[1, "A",
Point(9, 9),
datetime(2018, 1, 1, 14, 15, 0), 4, "d"],
[
2, "A",
Point(10, 10),
datetime(2018, 1, 1, 12, 0, 0), 10, "e"
],
[
2, "A",
Point(16, 10),
datetime(2018, 1, 1, 12, 6, 0), 6, "f"
],
[
2, "A",
Point(16, 16),
datetime(2018, 1, 2, 13, 10, 0), 7, "g"
],
[
2, "A",
Point(190, 10),
datetime(2018, 1, 2, 13, 15, 0), 3, "h"
],
],
columns=["id", "obj", "geometry", "t", "val", "val2"],
).set_index("t")
self.geo_df = GeoDataFrame(df, crs=CRS_METRIC)
self.collection = TrajectoryCollection(self.geo_df,
"id",
obj_id_col="obj")
self.geo_df_latlon = GeoDataFrame(df, crs=CRS_LATLON)
self.collection_latlon = TrajectoryCollection(self.geo_df_latlon,
"id",
obj_id_col="obj")
def setup_method(self):
self.nodes = [
Node(0, 0, day=1),
Node(1, 0.1, day=2),
Node(2, 0.2, day=3),
Node(3, 0, day=4),
Node(3, 3, day=5),
]
self.traj = make_traj(self.nodes)
df = pd.DataFrame([{
"xxx": n.geometry,
"t": n.t
} for n in self.nodes]).set_index("t")
geo_df = GeoDataFrame(df, geometry="xxx", crs=CRS_METRIC)
self.traj_other_geometry_column_names = Trajectory(geo_df, 1)
df = pd.DataFrame(
[
[1, "A",
Point(0, 0),
datetime(2018, 1, 1, 12, 0, 0), 9, "a"],
[1, "A",
Point(6, 0),
datetime(2018, 1, 1, 12, 6, 0), 5, "b"],
[1, "A",
Point(6, 6),
datetime(2018, 1, 1, 14, 10, 0), 2, "c"],
[1, "A",
Point(9, 9),
datetime(2018, 1, 1, 14, 15, 0), 4, "d"],
[
2, "A",
Point(10, 10),
datetime(2018, 1, 1, 12, 0, 0), 10, "e"
],
[
2, "A",
Point(16, 10),
datetime(2018, 1, 1, 12, 6, 0), 6, "f"
],
[
2, "A",
Point(16, 16),
datetime(2018, 1, 2, 13, 10, 0), 7, "g"
],
[
2, "A",
Point(190, 19),
datetime(2018, 1, 2, 13, 15, 0), 3, "h"
],
],
columns=["id", "obj", "geometry", "t", "val", "val2"],
).set_index("t")
self.geo_df = GeoDataFrame(df, crs=CRS_METRIC)
self.collection = TrajectoryCollection(self.geo_df,
"id",
obj_id_col="obj")
def test_traj_with_less_than_two_points(self):
df = pd.DataFrame(
[[1, "A",
Point(0, 0),
datetime(2018, 1, 1, 12, 0, 0), 9, "a"]],
columns=["id", "obj", "geometry", "t", "val", "val2"],
).set_index("t")
geo_df = GeoDataFrame(df, crs=CRS_METRIC)
tc = TrajectoryCollection(geo_df, "id", obj_id_col="obj")
assert len(tc) == 0
def setup_method(self):
df = pd.DataFrame([{
'id': 1,
'obj': 'A',
'geometry': Point(0, 0),
't': datetime(2018, 1, 1, 12, 0, 0),
'val': 9,
'val2': 'a'
}, {
'id': 1,
'obj': 'A',
'geometry': Point(6, 0),
't': datetime(2018, 1, 1, 12, 6, 0),
'val': 5,
'val2': 'b'
}, {
'id': 1,
'obj': 'A',
'geometry': Point(6, 6),
't': datetime(2018, 1, 1, 14, 10, 0),
'val': 2,
'val2': 'c'
}]).set_index('t')
self.geo_df = GeoDataFrame(df, crs=CRS_METRIC)
self.collection = TrajectoryCollection(self.geo_df,
'id',
obj_id_col='obj')
self.trajectory_aggregator = TrajectoryCollectionAggregator(
self.collection,
100,
0,
min_stop_duration=timedelta(hours=12).seconds)
self.geo_df_latlon = GeoDataFrame(df, crs=CRS_LATLON)
self.collection_latlon = TrajectoryCollection(self.geo_df_latlon,
'id',
obj_id_col='obj')
self.trajectory_aggregator_latlon = TrajectoryCollectionAggregator(
self.collection_latlon,
100,
0,
min_stop_duration=timedelta(hours=12).seconds)
def test_to_line_gdf(self):
temp_df = self.geo_df.drop(columns=["obj", "val", "val2"])
tc = TrajectoryCollection(temp_df, "id")
line_gdf = tc.to_line_gdf()
line_gdf.to_file("temp.gpkg", layer="lines", driver="GPKG")
t1 = [
datetime(2018, 1, 1, 12, 0),
datetime(2018, 1, 1, 12, 6),
datetime(2018, 1, 1, 14, 10),
datetime(2018, 1, 1, 14, 15),
]
t2 = [
datetime(2018, 1, 1, 12, 0, 0),
datetime(2018, 1, 1, 12, 6, 0),
datetime(2018, 1, 2, 13, 10, 0),
datetime(2018, 1, 2, 13, 15, 0),
]
df2 = pd.DataFrame(
[
[1, t1[1], t1[0],
LineString([(0, 0), (6, 0)])],
[1, t1[2], t1[1],
LineString([(6, 0), (6, 6)])],
[1, t1[3], t1[2],
LineString([(6, 6), (9, 9)])],
[2, t2[1], t2[0],
LineString([(10, 10), (16, 10)])],
[2, t2[2], t2[1],
LineString([(16, 10), (16, 16)])],
[2, t2[3], t2[2],
LineString([(16, 16), (190, 10)])],
],
columns=["id", "t", "prev_t", "geometry"],
)
expected_line_gdf = GeoDataFrame(df2, crs=CRS_METRIC)
assert_frame_equal(line_gdf, expected_line_gdf)
def test_stop_detector_no_stops(self):
traj1 = make_traj([
Node(0, 0),
Node(0, 10, second=10),
Node(0, 20, second=20),
Node(0, 30, second=30),
Node(0, 40, second=40),
Node(0, 50, second=50)
],
id=1)
collection = TrajectoryCollection([traj1])
detector = StopSplitter(collection)
stops = detector.split(max_diameter=1,
min_duration=timedelta(seconds=1))
assert len(stops) == 1
def setup_method(self):
self.pts = [
Point(0, 0),
Point(6, 0),
Point(6, 6),
Point(0.2, 0.2),
Point(6.2, 0.2),
Point(6.2, 6.2)
]
df = pd.DataFrame([{
'id': 1,
'obj': 'A',
'geometry': Point(0, 0),
't': datetime(2018, 1, 1, 12, 0, 0)
}, {
'id': 1,
'obj': 'A',
'geometry': Point(6, 0),
't': datetime(2018, 1, 1, 12, 6, 0)
}, {
'id': 1,
'obj': 'A',
'geometry': Point(6, 6),
't': datetime(2018, 1, 1, 14, 10, 0)
}, {
'id': 2,
'obj': 'A',
'geometry': Point(0.2, 0.2),
't': datetime(2019, 1, 1, 12, 0, 0)
}, {
'id': 2,
'obj': 'A',
'geometry': Point(6.2, 0.2),
't': datetime(2019, 1, 1, 12, 6, 0)
}, {
'id': 2,
'obj': 'A',
'geometry': Point(6.2, 6.2),
't': datetime(2019, 1, 1, 14, 10, 0)
}]).set_index('t')
self.geo_df = GeoDataFrame(df, crs=CRS_METRIC)
self.collection = TrajectoryCollection(self.geo_df,
'id',
obj_id_col='obj')
self.trajectory_aggregator = TrajectoryCollectionAggregator(
self.collection, 5, 0, min_stop_duration=timedelta(hours=12))
self.expected_clusters = GeoDataFrame(
pd.DataFrame([{
'geometry': Point(0.1, 0.1),
'n': 2
}, {
'geometry': Point(6.1, 0.1),
'n': 2
}, {
'geometry': Point(6.1, 6.1),
'n': 2
}]))
self.expected_flows = GeoDataFrame(
pd.DataFrame([{
'geometry':
LineString([Point(0.1, 0.1), Point(6.1, 0.1)]),
'weight':
2
}, {
'geometry':
LineString([Point(6.1, 0.1), Point(6.1, 6.1)]),
'weight':
2
}]))
self.geo_df_latlon = GeoDataFrame(df, crs=CRS_LATLON)
self.geo_df_latlon.crs = CRS_LATLON # try to fix https://travis-ci.com/github/anitagraser/movingpandas/builds/177149345
self.collection_latlon = TrajectoryCollection(self.geo_df_latlon,
'id',
obj_id_col='obj')
self.trajectory_aggregator_latlon = TrajectoryCollectionAggregator(
self.collection_latlon,
100,
0,
min_stop_duration=timedelta(hours=12))
def test_get_trajectory(self):
collection = TrajectoryCollection(self.geo_df, 'id', obj_id_col='obj')
assert collection.get_trajectory(1).id == 1
assert collection.get_trajectory(1).obj_id == 'A'
assert collection.get_trajectory(2).id == 2
assert collection.get_trajectory(3) is None
class TestTrajectoryCollection:
def setup_method(self):
df = pd.DataFrame([{
'id': 1,
'obj': 'A',
'geometry': Point(0, 0),
't': datetime(2018, 1, 1, 12, 0, 0),
'val': 9,
'val2': 'a'
}, {
'id': 1,
'obj': 'A',
'geometry': Point(6, 0),
't': datetime(2018, 1, 1, 12, 6, 0),
'val': 5,
'val2': 'b'
}, {
'id': 1,
'obj': 'A',
'geometry': Point(6, 6),
't': datetime(2018, 1, 1, 14, 10, 0),
'val': 2,
'val2': 'c'
}, {
'id': 1,
'obj': 'A',
'geometry': Point(9, 9),
't': datetime(2018, 1, 1, 14, 15, 0),
'val': 4,
'val2': 'd'
}, {
'id': 2,
'obj': 'A',
'geometry': Point(10, 10),
't': datetime(2018, 1, 1, 12, 0, 0),
'val': 10,
'val2': 'e'
}, {
'id': 2,
'obj': 'A',
'geometry': Point(16, 10),
't': datetime(2018, 1, 1, 12, 6, 0),
'val': 6,
'val2': 'f'
}, {
'id': 2,
'obj': 'A',
'geometry': Point(16, 16),
't': datetime(2018, 1, 2, 13, 10, 0),
'val': 7,
'val2': 'g'
}, {
'id': 2,
'obj': 'A',
'geometry': Point(190, 19),
't': datetime(2018, 1, 2, 13, 15, 0),
'val': 3,
'val2': 'h'
}]).set_index('t')
self.geo_df = GeoDataFrame(df, crs=CRS_METRIC)
self.collection = TrajectoryCollection(self.geo_df,
'id',
obj_id_col='obj')
self.geo_df_latlon = GeoDataFrame(df, crs=CRS_LATLON)
self.collection_latlon = TrajectoryCollection(self.geo_df_latlon,
'id',
obj_id_col='obj')
def test_number_of_trajectories(self):
assert len(self.collection) == 2
def test_number_of_trajectories_min_length(self):
collection = TrajectoryCollection(self.geo_df,
'id',
obj_id_col='obj',
min_length=100)
assert len(collection) == 1
def test_number_of_trajectories_min_length_never_reached(self):
collection = TrajectoryCollection(self.geo_df,
'id',
obj_id_col='obj',
min_length=1000)
assert len(collection) == 0
def test_get_trajectory(self):
assert self.collection.get_trajectory(1).id == 1
assert self.collection.get_trajectory(1).obj_id == 'A'
assert self.collection.get_trajectory(2).id == 2
assert self.collection.get_trajectory(3) is None
def test_get_start_locations(self):
locs = self.collection.get_start_locations(columns=['val', 'val2'])
assert len(locs) == 2
assert locs.iloc[0].geometry in [Point(0, 0), Point(10, 10)]
assert locs.iloc[0].traj_id in [1, 2]
assert locs.iloc[0].obj_id == 'A'
assert locs.iloc[0].val in [9, 10]
assert locs.iloc[0].val2 in ['a', 'e']
assert locs.iloc[1].geometry in [Point(0, 0), Point(10, 10)]
def test_get_end_locations(self):
locs = self.collection.get_end_locations(columns=['val', 'val2'])
assert len(locs) == 2
assert locs.iloc[0].geometry in [Point(9, 9), Point(190, 19)]
assert locs.iloc[0].traj_id in [1, 2]
assert locs.iloc[0].obj_id == 'A'
assert locs.iloc[0].val in [4, 3]
assert locs.iloc[0].val2 in ['d', 'h']
assert locs.iloc[1].geometry in [Point(9, 9), Point(190, 19)]
def test_generalize(self):
collection = self.collection.generalize(
mode='min-time-delta', tolerance=timedelta(minutes=10))
assert len(collection) == 2
assert collection.trajectories[0].to_linestring(
).wkt == 'LINESTRING (0 0, 6 6, 9 9)'
assert collection.trajectories[1].to_linestring(
).wkt == 'LINESTRING (10 10, 16 16, 190 19)'
def test_split_by_date(self):
collection = self.collection.split_by_date(mode='day')
assert len(collection) == 3
def test_split_by_observation_gap(self):
collection = self.collection.split_by_observation_gap(
timedelta(hours=1))
assert len(collection) == 4
def test_get_intersecting(self):
polygon = Polygon([(-1, -1), (-1, 1), (1, 1), (1, -1), (-1, -1)])
collection = self.collection.get_intersecting(polygon)
assert len(collection) == 1
assert collection.trajectories[0] == self.collection.trajectories[0]
def test_clip(self):
polygon = Polygon([(-1, -1), (-1, 1), (1, 1), (1, -1), (-1, -1)])
collection = self.collection.clip(polygon)
assert len(collection) == 1
assert collection.trajectories[0].to_linestring(
).wkt == 'LINESTRING (0 0, 1 0)'
def test_filter(self):
assert len(self.collection.filter('obj', 'A')) == 2
assert len(self.collection.filter('obj', 'B')) == 0
def test_get_min_and_max(self):
assert self.collection.get_min('val') == 2
assert self.collection.get_max('val') == 10
def test_plot_exists(self):
from matplotlib.axes import Axes
result = self.collection.plot()
assert isinstance(result, Axes)
def test_hvplot_exists(self):
import holoviews
result = self.collection_latlon.hvplot()
assert isinstance(result, holoviews.core.overlay.Overlay)
def test_number_of_trajectories_min_duration_never_reached(self):
collection = TrajectoryCollection(self.geo_df,
"id",
obj_id_col="obj",
min_duration=timedelta(weeks=1))
assert len(collection) == 0
def test_number_of_trajectories_min_duration_from_list(self):
collection = TrajectoryCollection(self.collection.trajectories,
min_duration=timedelta(days=1))
assert len(collection) == 1
def test_number_of_trajectories_min_length(self):
collection = TrajectoryCollection(self.geo_df,
"id",
obj_id_col="obj",
min_length=100)
assert len(collection) == 1
class TestTrajectoryCollection:
def setup_method(self):
df = pd.DataFrame([
{'id': 1, 'obj': 'A', 'geometry': Point(0, 0), 't': datetime(2018,1,1,12,0,0), 'val': 9},
{'id': 1, 'obj': 'A', 'geometry': Point(6, 0), 't': datetime(2018,1,1,12,6,0), 'val': 5},
{'id': 1, 'obj': 'A', 'geometry': Point(6, 6), 't': datetime(2018,1,1,12,10,0), 'val': 2},
{'id': 1, 'obj': 'A', 'geometry': Point(9, 9), 't': datetime(2018,1,1,12,15,0), 'val': 4},
{'id': 2, 'obj': 'A', 'geometry': Point(10, 10), 't': datetime(2018,1,1,12,0,0), 'val': 10},
{'id': 2, 'obj': 'A', 'geometry': Point(16, 10), 't': datetime(2018,1,1,12,6,0), 'val': 6},
{'id': 2, 'obj': 'A', 'geometry': Point(16, 16), 't': datetime(2018,1,2,12,10,0), 'val': 7},
{'id': 2, 'obj': 'A', 'geometry': Point(190, 19), 't': datetime(2018,1,2,12,15,0), 'val': 3}
]).set_index('t')
self.geo_df = GeoDataFrame(df, crs=CRS_METRIC)
self.collection = TrajectoryCollection(self.geo_df, 'id', obj_id_col='obj')
self.geo_df_latlon = GeoDataFrame(df, crs=CRS_LATLON)
self.collection_latlon = TrajectoryCollection(self.geo_df_latlon, 'id', obj_id_col='obj')
def test_number_of_trajectories(self):
assert len(self.collection) == 2
def test_number_of_trajectories_min_length(self):
collection = TrajectoryCollection(self.geo_df, 'id', obj_id_col='obj', min_length=100)
assert len(collection) == 1
def test_number_of_trajectories_min_length_never_reached(self):
collection = TrajectoryCollection(self.geo_df, 'id', obj_id_col='obj', min_length=1000)
assert len(collection) == 0
def test_split_by_date(self):
collection = self.collection.split_by_date(mode='day')
assert len(collection) == 3
def test_get_trajectory(self):
assert self.collection.get_trajectory(1).id == 1
assert self.collection.get_trajectory(1).obj_id == 'A'
assert self.collection.get_trajectory(2).id == 2
assert self.collection.get_trajectory(3) is None
def test_filter(self):
assert len(self.collection.filter('obj', 'A')) == 2
assert len(self.collection.filter('obj', 'B')) == 0
def test_get_min_and_max(self):
assert self.collection.get_min('val') == 2
assert self.collection.get_max('val') == 10
def test_get_start_locations(self):
locs = self.collection.get_start_locations(columns=['val'])
assert len(locs) == 2
assert locs.iloc[0].geometry in [Point(0, 0), Point(10, 10)]
assert locs.iloc[0].traj_id in [1, 2]
assert locs.iloc[0].obj_id == 'A'
assert locs.iloc[0].val in [9, 10]
assert locs.iloc[1].geometry in [Point(0, 0), Point(10, 10)]
def test_plot_exists(self):
from matplotlib.axes import Axes
result = self.collection.plot()
assert isinstance(result, Axes)
def test_hvplot_exists(self):
import holoviews
result = self.collection_latlon.hvplot()
assert isinstance(result, holoviews.core.overlay.Overlay)
def test_add_traj_id_overwrite_raises_error(self):
gdf = self.geo_df.copy()
gdf[TRAJ_ID_COL_NAME] = "a"
collection = TrajectoryCollection(gdf, "id", obj_id_col="obj")
with pytest.raises(RuntimeError):
collection.add_traj_id()
def test_split_by_date(self):
collection = TrajectoryCollection(self.geo_df, 'id', obj_id_col='obj')
collection = collection.split_by_date(mode='day')
assert len(collection) == 3
class TestTrajectoryCollection:
def setup_method(self):
df = pd.DataFrame([{
'id': 1,
'obj': 'A',
'geometry': Point(0, 0),
't': datetime(2018, 1, 1, 12, 0, 0),
'val': 9,
'val2': 'a'
}, {
'id': 1,
'obj': 'A',
'geometry': Point(6, 0),
't': datetime(2018, 1, 1, 12, 6, 0),
'val': 5,
'val2': 'b'
}, {
'id': 1,
'obj': 'A',
'geometry': Point(6, 6),
't': datetime(2018, 1, 1, 14, 10, 0),
'val': 2,
'val2': 'c'
}, {
'id': 1,
'obj': 'A',
'geometry': Point(9, 9),
't': datetime(2018, 1, 1, 14, 15, 0),
'val': 4,
'val2': 'd'
}, {
'id': 2,
'obj': 'A',
'geometry': Point(10, 10),
't': datetime(2018, 1, 1, 12, 0, 0),
'val': 10,
'val2': 'e'
}, {
'id': 2,
'obj': 'A',
'geometry': Point(16, 10),
't': datetime(2018, 1, 1, 12, 6, 0),
'val': 6,
'val2': 'f'
}, {
'id': 2,
'obj': 'A',
'geometry': Point(16, 16),
't': datetime(2018, 1, 2, 13, 10, 0),
'val': 7,
'val2': 'g'
}, {
'id': 2,
'obj': 'A',
'geometry': Point(190, 19),
't': datetime(2018, 1, 2, 13, 15, 0),
'val': 3,
'val2': 'h'
}]).set_index('t')
self.geo_df = GeoDataFrame(df, crs=CRS_METRIC)
self.collection = TrajectoryCollection(self.geo_df,
'id',
obj_id_col='obj')
self.geo_df_latlon = GeoDataFrame(df, crs=CRS_LATLON)
self.collection_latlon = TrajectoryCollection(self.geo_df_latlon,
'id',
obj_id_col='obj')
def test_number_of_trajectories(self):
assert len(self.collection) == 2
def test_number_of_trajectories_min_length(self):
collection = TrajectoryCollection(self.geo_df,
'id',
obj_id_col='obj',
min_length=100)
assert len(collection) == 1
def test_number_of_trajectories_min_length_never_reached(self):
collection = TrajectoryCollection(self.geo_df,
'id',
obj_id_col='obj',
min_length=1000)
assert len(collection) == 0
def test_get_trajectory(self):
assert self.collection.get_trajectory(1).id == 1
assert self.collection.get_trajectory(1).obj_id == 'A'
assert self.collection.get_trajectory(2).id == 2
assert self.collection.get_trajectory(3) is None
def test_get_locations_at(self):
locs = self.collection.get_locations_at(datetime(2018, 1, 1, 12, 6, 0))
assert len(locs) == 2
assert locs.iloc[0].geometry in [Point(6, 0), Point(16, 10)]
assert locs.iloc[0].val in [5, 6]
assert locs.iloc[1].geometry in [Point(6, 0), Point(16, 10)]
assert locs.iloc[0].geometry != locs.iloc[1].geometry
def test_get_locations_at_needing_interpolation(self):
locs = self.collection.get_locations_at(datetime(2018, 1, 1, 12, 6, 1))
assert len(locs) == 2
assert locs.iloc[0].val in [5, 6]
def test_get_locations_at_out_of_time_range(self):
locs = self.collection.get_locations_at(datetime(2017, 1, 1, 12, 6, 1))
assert len(locs) == 0
def test_get_start_locations(self):
locs = self.collection.get_start_locations()
assert len(locs) == 2
assert locs.iloc[0].geometry in [Point(0, 0), Point(10, 10)]
assert locs.iloc[0].id in [1, 2]
assert locs.iloc[0].obj == 'A'
assert locs.iloc[0].val in [9, 10]
assert locs.iloc[0].val2 in ['a', 'e']
assert locs.iloc[1].geometry in [Point(0, 0), Point(10, 10)]
assert locs.iloc[0].geometry != locs.iloc[1].geometry
def test_get_end_locations(self):
locs = self.collection.get_end_locations()
assert len(locs) == 2
assert locs.iloc[0].geometry in [Point(9, 9), Point(190, 19)]
assert locs.iloc[0].id in [1, 2]
assert locs.iloc[0].obj == 'A'
assert locs.iloc[0].val in [4, 3]
assert locs.iloc[0].val2 in ['d', 'h']
assert locs.iloc[1].geometry in [Point(9, 9), Point(190, 19)]
assert locs.iloc[0].geometry != locs.iloc[1].geometry
def test_get_intersecting(self):
polygon = Polygon([(-1, -1), (-1, 1), (1, 1), (1, -1), (-1, -1)])
collection = self.collection.get_intersecting(polygon)
assert len(collection) == 1
assert collection.trajectories[0] == self.collection.trajectories[0]
def test_clip(self):
polygon = Polygon([(-1, -1), (-1, 1), (1, 1), (1, -1), (-1, -1)])
collection = self.collection.clip(polygon)
assert len(collection) == 1
assert collection.trajectories[0].to_linestring(
).wkt == 'LINESTRING (0 0, 1 0)'
def test_filter(self):
assert len(self.collection.filter('obj', 'A')) == 2
assert len(self.collection.filter('obj', ['A'])) == 2
assert len(self.collection.filter('obj', ['B'])) == 0
assert len(self.collection.filter('obj', [1])) == 0
def test_get_min_and_max(self):
assert self.collection.get_min('val') == 2
assert self.collection.get_max('val') == 10
def test_plot_exists(self):
from matplotlib.axes import Axes
result = self.collection.plot()
assert isinstance(result, Axes)
def test_hvplot_exists(self):
import holoviews
result = self.collection_latlon.hvplot()
assert isinstance(result, holoviews.core.overlay.Overlay)
def test_plot_exist_column(self):
from matplotlib.axes import Axes
result = self.collection.plot(column='val')
assert isinstance(result, Axes)
def test_plot_speed_not_altering_collection(self):
self.collection.plot(column='speed')
assert (all([
'speed' not in traj.df.columns.values
for traj in self.collection.trajectories
]))
def test_traj_with_less_than_two_points(self):
df = pd.DataFrame([{
'id': 1,
'obj': 'A',
'geometry': Point(0, 0),
't': datetime(2018, 1, 1, 12, 0, 0),
'val': 9,
'val2': 'a'
}]).set_index('t')
geo_df = GeoDataFrame(df, crs=CRS_METRIC)
tc = TrajectoryCollection(geo_df, 'id', obj_id_col='obj')
assert len(tc) == 0
def test_traj_with_two_points_at_the_same_time(self):
df = pd.DataFrame([
{
'id': 1,
'obj': 'A',
'geometry': Point(0, 0),
't': datetime(2018, 1, 1, 12, 0, 0),
'val': 9,
'val2': 'a'
},
{
'id': 1,
'obj': 'A',
'geometry': Point(0, 0),
't': datetime(2018, 1, 1, 12, 0, 0),
'val': 9,
'val2': 'a'
},
]).set_index('t')
geo_df = GeoDataFrame(df, crs=CRS_METRIC)
tc = TrajectoryCollection(geo_df, 'id', obj_id_col='obj')
assert len(tc) == 0
def test_iteration(self):
assert sum([1 for _ in self.collection]) == len(self.collection)
def test_iteration_error(self):
def filter_trajectory(trajectory):
trajectory.df = trajectory.df[trajectory.df['val'] >= 7]
return trajectory
trajs = [filter_trajectory(traj) for traj in self.collection]
lengths = (1, 2)
for i, traj in enumerate(trajs):
assert len(traj.df) == lengths[i]
collection = copy(self.collection)
collection.trajectories = trajs
with pytest.raises(ValueError):
for _ in collection:
pass
class TestTrajectoryCollection:
def setup_method(self):
df = pd.DataFrame(
[
[1, "A",
Point(0, 0),
datetime(2018, 1, 1, 12, 0, 0), 9, "a"],
[1, "A",
Point(6, 0),
datetime(2018, 1, 1, 12, 6, 0), 5, "b"],
[1, "A",
Point(6, 6),
datetime(2018, 1, 1, 14, 10, 0), 2, "c"],
[1, "A",
Point(9, 9),
datetime(2018, 1, 1, 14, 15, 0), 4, "d"],
[
2, "A",
Point(10, 10),
datetime(2018, 1, 1, 12, 0, 0), 10, "e"
],
[
2, "A",
Point(16, 10),
datetime(2018, 1, 1, 12, 6, 0), 6, "f"
],
[
2, "A",
Point(16, 16),
datetime(2018, 1, 2, 13, 10, 0), 7, "g"
],
[
2, "A",
Point(190, 10),
datetime(2018, 1, 2, 13, 15, 0), 3, "h"
],
],
columns=["id", "obj", "geometry", "t", "val", "val2"],
).set_index("t")
self.geo_df = GeoDataFrame(df, crs=CRS_METRIC)
self.collection = TrajectoryCollection(self.geo_df,
"id",
obj_id_col="obj")
self.geo_df_latlon = GeoDataFrame(df, crs=CRS_LATLON)
self.collection_latlon = TrajectoryCollection(self.geo_df_latlon,
"id",
obj_id_col="obj")
def test_number_of_trajectories(self):
assert len(self.collection) == 2
def test_number_of_trajectories_min_length(self):
collection = TrajectoryCollection(self.geo_df,
"id",
obj_id_col="obj",
min_length=100)
assert len(collection) == 1
def test_number_of_trajectories_min_length_never_reached(self):
collection = TrajectoryCollection(self.geo_df,
"id",
obj_id_col="obj",
min_length=1000)
assert len(collection) == 0
def test_number_of_trajectories_min_duration(self):
collection = TrajectoryCollection(self.geo_df,
"id",
obj_id_col="obj",
min_duration=timedelta(days=1))
assert len(collection) == 1
def test_number_of_trajectories_min_duration_from_list(self):
collection = TrajectoryCollection(self.collection.trajectories,
min_duration=timedelta(days=1))
assert len(collection) == 1
def test_number_of_trajectories_min_duration_never_reached(self):
collection = TrajectoryCollection(self.geo_df,
"id",
obj_id_col="obj",
min_duration=timedelta(weeks=1))
assert len(collection) == 0
def test_get_trajectory(self):
assert self.collection.get_trajectory(1).id == 1
assert self.collection.get_trajectory(1).obj_id == "A"
assert self.collection.get_trajectory(2).id == 2
assert self.collection.get_trajectory(3) is None
def test_get_locations_at(self):
locs = self.collection.get_locations_at(datetime(2018, 1, 1, 12, 6, 0))
assert len(locs) == 2
assert locs.iloc[0].geometry in [Point(6, 0), Point(16, 10)]
assert locs.iloc[0].val in [5, 6]
assert locs.iloc[1].geometry in [Point(6, 0), Point(16, 10)]
assert locs.iloc[0].geometry != locs.iloc[1].geometry
def test_get_locations_at_needing_interpolation(self):
locs = self.collection.get_locations_at(datetime(2018, 1, 1, 12, 6, 1))
assert len(locs) == 2
assert locs.iloc[0].val in [5, 6]
def test_get_locations_at_out_of_time_range(self):
locs = self.collection.get_locations_at(datetime(2017, 1, 1, 12, 6, 1))
assert len(locs) == 0
def test_get_start_locations(self):
locs = self.collection.get_start_locations()
assert len(locs) == 2
assert locs.iloc[0].geometry in [Point(0, 0), Point(10, 10)]
assert locs.iloc[0].id in [1, 2]
assert locs.iloc[0].obj == "A"
assert locs.iloc[0].val in [9, 10]
assert locs.iloc[0].val2 in ["a", "e"]
assert locs.iloc[1].geometry in [Point(0, 0), Point(10, 10)]
assert locs.iloc[0].geometry != locs.iloc[1].geometry
assert isinstance(locs, GeoDataFrame)
def test_get_end_locations(self):
locs = self.collection.get_end_locations()
assert len(locs) == 2
assert locs.iloc[0].geometry in [Point(9, 9), Point(190, 10)]
assert locs.iloc[0].id in [1, 2]
assert locs.iloc[0].obj == "A"
assert locs.iloc[0].val in [4, 3]
assert locs.iloc[0].val2 in ["d", "h"]
assert locs.iloc[1].geometry in [Point(9, 9), Point(190, 10)]
assert locs.iloc[0].geometry != locs.iloc[1].geometry
assert isinstance(locs, GeoDataFrame)
def test_get_intersecting(self):
polygon = Polygon([(-1, -1), (-1, 1), (1, 1), (1, -1), (-1, -1)])
collection = self.collection.get_intersecting(polygon)
assert len(collection) == 1
assert collection.trajectories[0] == self.collection.trajectories[0]
def test_clip(self):
polygon = Polygon([(-1, -1), (-1, 1), (1, 1), (1, -1), (-1, -1)])
collection = self.collection.clip(polygon)
assert len(collection) == 1
assert collection.trajectories[0].to_linestring(
).wkt == "LINESTRING (0 0, 1 0)"
def test_filter(self):
assert len(self.collection.filter("obj", "A")) == 2
assert len(self.collection.filter("obj", ["A"])) == 2
assert len(self.collection.filter("obj", ["B"])) == 0
assert len(self.collection.filter("obj", [1])) == 0
def test_get_min_and_max(self):
assert self.collection.get_min("val") == 2
assert self.collection.get_max("val") == 10
def test_plot_exists(self):
from matplotlib.axes import Axes
result = self.collection.plot()
assert isinstance(result, Axes)
def test_hvplot_exists(self):
import holoviews
result = self.collection_latlon.hvplot()
assert isinstance(result, holoviews.core.overlay.Overlay)
def test_plot_exist_column(self):
from matplotlib.axes import Axes
result = self.collection.plot(column="val")
assert isinstance(result, Axes)
def test_plot_speed_not_altering_collection(self):
self.collection.plot(column="speed")
assert all([
"speed" not in traj.df.columns.values
for traj in self.collection.trajectories
])
def test_traj_with_less_than_two_points(self):
df = pd.DataFrame(
[[1, "A",
Point(0, 0),
datetime(2018, 1, 1, 12, 0, 0), 9, "a"]],
columns=["id", "obj", "geometry", "t", "val", "val2"],
).set_index("t")
geo_df = GeoDataFrame(df, crs=CRS_METRIC)
tc = TrajectoryCollection(geo_df, "id", obj_id_col="obj")
assert len(tc) == 0
def test_traj_with_two_points_at_the_same_time(self):
df = pd.DataFrame(
[
[1, "A",
Point(0, 0),
datetime(2018, 1, 1, 12, 0, 0), 9, "a"],
[1, "A",
Point(0, 0),
datetime(2018, 1, 1, 12, 0, 0), 9, "a"],
],
columns=["id", "obj", "geometry", "t", "val", "val2"],
).set_index("t")
geo_df = GeoDataFrame(df, crs=CRS_METRIC)
tc = TrajectoryCollection(geo_df, "id", obj_id_col="obj")
assert len(tc) == 0
def test_iteration(self):
assert sum([1 for _ in self.collection]) == len(self.collection)
def test_iteration_error(self):
def filter_trajectory(trajectory):
trajectory.df = trajectory.df[trajectory.df["val"] >= 7]
return trajectory
trajs = [filter_trajectory(traj) for traj in self.collection]
lengths = (1, 2)
for i, traj in enumerate(trajs):
assert len(traj.df) == lengths[i]
collection = copy(self.collection)
collection.trajectories = trajs
with pytest.raises(ValueError):
for _ in collection:
pass
def test_add_traj_id(self):
self.collection.add_traj_id()
result1 = self.collection.trajectories[0].df[TRAJ_ID_COL_NAME].tolist()
assert result1 == [1, 1, 1, 1]
result2 = self.collection.trajectories[1].df[TRAJ_ID_COL_NAME].tolist()
assert result2 == [2, 2, 2, 2]
def test_add_traj_id_overwrite_raises_error(self):
gdf = self.geo_df.copy()
gdf[TRAJ_ID_COL_NAME] = "a"
collection = TrajectoryCollection(gdf, "id", obj_id_col="obj")
with pytest.raises(RuntimeError):
collection.add_traj_id()
def test_to_point_gdf(self):
point_gdf = self.collection.to_point_gdf()
point_gdf.to_file("temp.gpkg", layer="points", driver="GPKG")
assert_frame_equal(point_gdf, self.geo_df)
def test_to_line_gdf(self):
temp_df = self.geo_df.drop(columns=["obj", "val", "val2"])
tc = TrajectoryCollection(temp_df, "id")
line_gdf = tc.to_line_gdf()
line_gdf.to_file("temp.gpkg", layer="lines", driver="GPKG")
t1 = [
datetime(2018, 1, 1, 12, 0),
datetime(2018, 1, 1, 12, 6),
datetime(2018, 1, 1, 14, 10),
datetime(2018, 1, 1, 14, 15),
]
t2 = [
datetime(2018, 1, 1, 12, 0, 0),
datetime(2018, 1, 1, 12, 6, 0),
datetime(2018, 1, 2, 13, 10, 0),
datetime(2018, 1, 2, 13, 15, 0),
]
df2 = pd.DataFrame(
[
[1, t1[1], t1[0],
LineString([(0, 0), (6, 0)])],
[1, t1[2], t1[1],
LineString([(6, 0), (6, 6)])],
[1, t1[3], t1[2],
LineString([(6, 6), (9, 9)])],
[2, t2[1], t2[0],
LineString([(10, 10), (16, 10)])],
[2, t2[2], t2[1],
LineString([(16, 10), (16, 16)])],
[2, t2[3], t2[2],
LineString([(16, 16), (190, 10)])],
],
columns=["id", "t", "prev_t", "geometry"],
)
expected_line_gdf = GeoDataFrame(df2, crs=CRS_METRIC)
assert_frame_equal(line_gdf, expected_line_gdf)
def test_to_traj_gdf(self):
temp_df = self.geo_df.drop(columns=["obj", "val", "val2"])
tc = TrajectoryCollection(temp_df, "id")
traj_gdf = tc.to_traj_gdf()
traj_gdf.to_file("temp.gpkg", layer="trajs", driver="GPKG")
rows = [
{
"traj_id": 1,
"start_t": datetime(2018, 1, 1, 12, 0, 0),
"end_t": datetime(2018, 1, 1, 14, 15, 0),
"geometry": LineString([(0, 0), (6, 0), (6, 6), (9, 9)]),
"length": 12 + sqrt(18),
"direction": 45.0,
},
{
"traj_id": 2,
"start_t": datetime(2018, 1, 1, 12, 0, 0),
"end_t": datetime(2018, 1, 2, 13, 15, 0),
"geometry": LineString([(10, 10), (16, 10), (16, 16),
(190, 10)]),
"length": 12 + sqrt(174 * 174 + 36),
"direction": 90.0,
},
]
df2 = pd.DataFrame(rows)
expected_line_gdf = GeoDataFrame(df2, crs=CRS_METRIC)
assert_frame_equal(traj_gdf, expected_line_gdf)
def test_filter(self):
collection = TrajectoryCollection(self.geo_df, 'id', obj_id_col='obj')
assert len(collection.filter('obj', 'A')) == 2
assert len(collection.filter('obj', 'B')) == 0
def test_get_min_and_max(self):
collection = TrajectoryCollection(self.geo_df, 'id', obj_id_col='obj')
assert collection.get_min('val') == 2
assert collection.get_max('val') == 10
def test_number_of_trajectories_min_length_never_reached(self):
collection = TrajectoryCollection(self.geo_df,
'id',
obj_id_col='obj',
min_length=1000)
assert len(collection) == 0
def test_number_of_trajectories(self):
collection = TrajectoryCollection(self.geo_df, 'id', obj_id_col='obj')
assert len(collection) == 2
