def test_trajectory_bounding_boxes_zeros():
result = cuspatial.trajectory_bounding_boxes(1, cudf.Series([0]),
cudf.Series([0]),
cudf.Series([0]))
cudf.testing.assert_frame_equal(
result,
cudf.DataFrame({
"x_min": [0.0],
"y_min": [0.0],
"x_max": [0.0],
"y_max": [0.0]
}),
)
def test_trajectory_bounding_boxes_ones():
result = cuspatial.trajectory_bounding_boxes(1, cudf.Series([1]),
cudf.Series([1]),
cudf.Series([1]))
assert_eq(
result,
cudf.DataFrame({
"x_min": [1.0],
"y_min": [1.0],
"x_max": [1.0],
"y_max": [1.0]
}),
)
def test_trajectory_bounding_boxes_zero_to_one_xy():
result = cuspatial.trajectory_bounding_boxes(
1,
cudf.Series([0, 0]),
cudf.Series([0, 1]),
cudf.Series([0, 1]),
)
assert_eq(
result,
cudf.DataFrame({
"x_min": [0.0],
"y_min": [0.0],
"x_max": [1.0],
"y_max": [1.0]
}),
)
def test_trajectory_bounding_boxes_empty_float64():
result = cuspatial.trajectory_bounding_boxes(
0,
cudf.Series(),
cudf.Series([], dtype=np.float64),
cudf.Series([], dtype=np.float64),
)
assert_eq(
result,
cudf.DataFrame({
"x_min": cudf.Series([], dtype=np.float64),
"y_min": cudf.Series([], dtype=np.float64),
"x_max": cudf.Series([], dtype=np.float64),
"y_max": cudf.Series([], dtype=np.float64),
}),
)
def test_trajectory_bounding_boxes_two_and_three():
result = cuspatial.trajectory_bounding_boxes(
2,
cudf.Series([0, 0, 1, 1, 1]),
cudf.Series([0, 2, 1, 3, 2]),
cudf.Series([0, 2, 1, 3, 2]),
)
assert_eq(
result,
cudf.DataFrame({
"x_min": [0.0, 1.0],
"y_min": [0.0, 1.0],
"x_max": [2.0, 3.0],
"y_max": [2.0, 3.0],
}),
)
def test_trajectory_bounding_boxes_subsetted():
result = cuspatial.trajectory_bounding_boxes(
2,
cudf.Series([0, 0, 1, 1]),
cudf.Series([0, 1, -1, 2]),
cudf.Series([0, 1, -1, 2]),
)
assert_eq(
result,
cudf.DataFrame({
"x_min": [0.0, -1.0],
"y_min": [0.0, -1.0],
"x_max": [1.0, 2.0],
"y_max": [1.0, 2.0],
}),
)
def test_trajectory_bounding_boxes_intersected():
result = cuspatial.trajectory_bounding_boxes(
2,
cudf.Series([0, 0, 1, 1]),
cudf.Series([0, 2, 1, 3]),
cudf.Series([0, 2, 1, 3]),
)
cudf.testing.assert_frame_equal(
result,
cudf.DataFrame({
"x_min": [0.0, 1.0],
"y_min": [0.0, 1.0],
"x_max": [2.0, 3.0],
"y_max": [2.0, 3.0],
}),
)
def test_trajectory_bounding_boxes_empty_float32():
result = cuspatial.trajectory_bounding_boxes(
0,
cudf.Series(),
cudf.Series([], dtype=np.float32),
cudf.Series([], dtype=np.float32),
)
cudf.testing.assert_frame_equal(
result,
cudf.DataFrame({
"x_min": cudf.Series([], dtype=np.float32),
"y_min": cudf.Series([], dtype=np.float32),
"x_max": cudf.Series([], dtype=np.float32),
"y_max": cudf.Series([], dtype=np.float32),
}),
)
