Function verify_shapes

tests/test_parse.py:255–337

(
    true_shapes: List[PdfShape], pred_shapes: List[PdfShape], eps: float
)

def verify_shapes(
    true_shapes: List[PdfShape], pred_shapes: List[PdfShape], eps: float
) -> bool:

    assert len(true_shapes) == len(pred_shapes), "len(true_shapes)==len(pred_shapes)"

    for i, true_shape in enumerate(true_shapes):
        pred_shape = pred_shapes[i]

        assert true_shape.index == pred_shape.index, (
            "true_shape.index == pred_shape.index"
        )

        assert true_shape.parent_id == pred_shape.parent_id, (
            "true_shape.parent_id == pred_shape.parent_id"
        )

        true_points = true_shape.points
        pred_points = pred_shape.points

        assert len(true_points) == len(pred_points), (
            "len(true_points) == len(pred_points)"
        )

        for point_idx, true_point in enumerate(true_points):
            assert abs(true_point[0] - pred_points[point_idx][0]) < eps, (
                "abs(true_point[0]-pred_points[l][0])<eps"
            )
            assert abs(true_point[1] - pred_points[point_idx][1]) < eps, (
                "abs(true_point[1]-pred_points[l][1])<eps"
            )

        assert true_shape.has_graphics_state == pred_shape.has_graphics_state, (
            "true_shape.has_graphics_state == pred_shape.has_graphics_state"
        )

        assert abs(true_shape.line_width - pred_shape.line_width) < eps, (
            "abs(true_shape.line_width - pred_shape.line_width) < eps"
        )
        assert abs(true_shape.miter_limit - pred_shape.miter_limit) < eps, (
            "abs(true_shape.miter_limit - pred_shape.miter_limit) < eps"
        )
        assert true_shape.line_cap == pred_shape.line_cap, (
            "true_shape.line_cap == pred_shape.line_cap"
        )
        assert true_shape.line_join == pred_shape.line_join, (
            "true_shape.line_join == pred_shape.line_join"
        )
        assert abs(true_shape.dash_phase - pred_shape.dash_phase) < eps, (
            "abs(true_shape.dash_phase - pred_shape.dash_phase) < eps"
        )
        assert len(true_shape.dash_array) == len(pred_shape.dash_array), (
            "len(true_shape.dash_array) == len(pred_shape.dash_array)"
        )
        for j, true_dash in enumerate(true_shape.dash_array):
            assert abs(true_dash - pred_shape.dash_array[j]) < eps, (
                "abs(true_dash - pred_shape.dash_array[j]) < eps"
            )