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), }), )