def field_data(bounding_box, form_word, form_line):
model = _models.FieldData(page_number=1,
text="This is text.",
bounding_box=bounding_box[0],
field_elements=[form_word[0], form_line[0]])
model_repr = "FieldData(page_number=1, text=This is text., bounding_box={}, field_elements=[{}, {}])".format(
bounding_box[1], form_word[1], form_line[1])[:1024]
assert repr(model) == model_repr
return model, model_repr
def test_field_data_to_dict(self):
model = _models.FieldData(
text="element",
page_number=1,
bounding_box=[
_models.Point(1427.0, 1669.0),
_models.Point(1527.0, 1669.0),
_models.Point(1527.0, 1698.0),
_models.Point(1427.0, 1698.0),
],
field_elements=[
_models.FormWord(
text="word",
confidence=0.92,
page_number=1,
bounding_box=[
_models.Point(1427.0, 1669.0),
_models.Point(1527.0, 1669.0),
_models.Point(1527.0, 1698.0),
_models.Point(1427.0, 1698.0),
],
),
],
)
d = model.to_dict()
final = {
"text": "element",
"bounding_box": [
{"x": 1427.0, "y": 1669.0},
{"x": 1527.0, "y": 1669.0},
{"x": 1527.0, "y": 1698.0},
{"x": 1427.0, "y": 1698.0},
],
"page_number": 1,
"field_elements": [
{
"text": "word",
"bounding_box": [
{"x": 1427.0, "y": 1669.0},
{"x": 1527.0, "y": 1669.0},
{"x": 1527.0, "y": 1698.0},
{"x": 1427.0, "y": 1698.0},
],
"confidence": 0.92,
"page_number": 1,
"kind": "word",
}
],
}
assert d == final
