1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
|
# coding=utf-8
# --------------------------------------------------------------------------
# Copyright (c) Microsoft Corporation. All rights reserved.
# Licensed under the MIT License. See License.txt in the project root for
# license information.
# --------------------------------------------------------------------------
import pytest
import datetime
from azure.ai.formrecognizer import _models
from azure.ai.formrecognizer import Appearance, Style
# All features return a tuple of the object and the repr of the obejct
# Adding in assert for each pytest fixture so it's easier to narrow down where the problem is
@pytest.fixture
def bounding_box():
model = [
_models.Point(1, 2),
_models.Point(3, 4),
_models.Point(5, 6),
_models.Point(7, 8)
]
model_repr = '[Point(x=1, y=2), Point(x=3, y=4), Point(x=5, y=6), Point(x=7, y=8)]'
assert repr(model) == model_repr
return model, model_repr
@pytest.fixture
def form_word(bounding_box):
model = _models.FormWord(text="Word", bounding_box=bounding_box[0], confidence=0.5, page_number=1)
model_repr = "FormWord(text=Word, bounding_box={}, confidence=0.5, page_number=1, kind=word)".format(bounding_box[1])[:1024]
assert repr(model) == model_repr
return model, model_repr
@pytest.fixture
def form_line(bounding_box, form_word):
appearance = Appearance(style=Style(name="other", confidence=1.0))
model = _models.FormLine(text="Word Word", bounding_box=bounding_box[0], words=[form_word[0], form_word[0]], page_number=1, appearance=appearance)
model_repr = "FormLine(text=Word Word, bounding_box={}, words=[{}, {}], page_number=1, kind=line, appearance={})".format(bounding_box[1], form_word[1], form_word[1], appearance)[:1024]
assert repr(model) == model_repr
return model, model_repr
@pytest.fixture
def form_table_cell(bounding_box, form_word):
model = _models.FormTableCell(
text="Cell", row_index=3, column_index=4, row_span=2, column_span=3, bounding_box=bounding_box[0],
confidence=0.7, is_header=True, is_footer=False, page_number=3, field_elements=[form_word[0]]
)
model_repr = "FormTableCell(text=Cell, row_index=3, column_index=4, row_span=2, column_span=3, bounding_box={}, confidence=0.7, " \
"is_header=True, is_footer=False, page_number=3, field_elements=[{}])".format(bounding_box[1], form_word[1])[:1024]
assert repr(model) == model_repr
return model, model_repr
@pytest.fixture
def form_table(form_table_cell, bounding_box):
model = _models.FormTable(page_number=1, cells=[form_table_cell[0], form_table_cell[0]], row_count=3, column_count=4, bounding_box=bounding_box[0])
model_repr = "FormTable(page_number=1, cells=[{}, {}], row_count=3, column_count=4, bounding_box={})".format(form_table_cell[1], form_table_cell[1], bounding_box[1])[:1024]
assert repr(model) == model_repr
return model, model_repr
@pytest.fixture
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
@pytest.fixture
def form_field_two(field_data):
model = _models.FormField(value_type="string", label_data=field_data[0], value_data=field_data[0], name="form_field_two", value="value", confidence=0)
model_repr = "FormField(value_type=string, label_data={}, value_data={}, name=form_field_two, value='value', confidence=0)".format(field_data[1], field_data[1])[:1024]
assert repr(model) == model_repr
return model, model_repr
@pytest.fixture
def form_field_one(field_data, form_field_two):
model = _models.FormField(value_type="string", label_data=field_data[0], value_data=field_data[0], name="form_field_one", value=form_field_two[0], confidence=1.0)
model_repr = "FormField(value_type=string, label_data={}, value_data={}, name=form_field_one, value={}, confidence=1.0)".format(field_data[1], field_data[1], form_field_two[1])[:1024]
assert repr(model) == model_repr
return model, model_repr
@pytest.fixture
def page_range():
model = _models.FormPageRange(first_page_number=1, last_page_number=100)
model_repr = "FormPageRange(first_page_number=1, last_page_number=100)"
assert repr(model) == model_repr
return model, model_repr
@pytest.fixture
def form_page(form_table, form_line):
model = _models.FormPage(page_number=1, text_angle=180, width=5, height=5.5, unit=_models.LengthUnit.PIXEL, tables=[form_table[0]], lines=[form_line[0]])
model_repr = "FormPage(page_number=1, text_angle=180, width=5, height=5.5, unit=pixel, tables=[{}], lines=[{}])".format(
form_table[1], form_line[1]
)[:1024]
assert repr(model) == model_repr
return model, model_repr
@pytest.fixture
def custom_form_model_field():
model = _models.CustomFormModelField(label="label", name="name", accuracy=0.99)
model_repr = "CustomFormModelField(label=label, name=name, accuracy=0.99)"
assert repr(model) == model_repr
return model, model_repr
@pytest.fixture
def custom_form_sub_model(custom_form_model_field):
model = _models.CustomFormSubmodel(accuracy=0.99, model_id=1, fields={"name": custom_form_model_field[0]}, form_type="Itemized")
model_repr = "CustomFormSubmodel(accuracy=0.99, model_id=1, fields={{'name': {}}}, form_type=Itemized)".format(custom_form_model_field[1])[:1024]
assert repr(model) == model_repr
return model, model_repr
@pytest.fixture
def form_recognizer_error():
model = _models.FormRecognizerError(code=404, message="Resource Not Found")
model_repr = "FormRecognizerError(code=404, message=Resource Not Found)"
assert repr(model) == model_repr
return model, model_repr
@pytest.fixture
def training_document_info(form_recognizer_error):
model = _models.TrainingDocumentInfo(name="name", status=_models.TrainingStatus.PARTIALLY_SUCCEEDED, page_count=5, errors=[form_recognizer_error[0]], model_id=1)
model_repr = "TrainingDocumentInfo(name=name, status=partiallySucceeded, page_count=5, errors=[{}], model_id=1)".format(form_recognizer_error[1])[:1024]
assert repr(model) == model_repr
return model, model_repr
@pytest.fixture
def custom_form_model_properties():
model = _models.CustomFormModelProperties(is_composed_model=True)
model_repr = "CustomFormModelProperties(is_composed_model=True)"
assert repr(model) == model_repr
return model, model_repr
class TestRepr():
# Not inheriting form FormRecognizerTest because that doesn't allow me to define pytest fixtures in the same file
# Not worth moving pytest fixture definitions to conftest since all I would use is assertEqual and I can just use assert
def test_recognized_form(self, form_field_one, page_range, form_page):
model = _models.RecognizedForm(form_type="receipt", form_type_confidence=1.0, model_id=1, fields={"one": form_field_one[0]}, page_range=page_range[0], pages=[form_page[0]])
model_repr = "RecognizedForm(form_type=receipt, fields={{'one': {}}}, page_range={}, pages=[{}], form_type_confidence=1.0, model_id=1)".format(
form_field_one[1], page_range[1], form_page[1]
)[:1024]
assert repr(model) == model_repr
def test_custom_form_model(self, custom_form_sub_model, custom_form_model_properties, form_recognizer_error, training_document_info):
model = _models.CustomFormModel(
model_id=1,
status=_models.CustomFormModelStatus.CREATING,
training_started_on=datetime.datetime(1, 1, 1),
training_completed_on=datetime.datetime(1, 1, 1),
submodels=[custom_form_sub_model[0], custom_form_sub_model[0]],
errors=[form_recognizer_error[0]],
training_documents=[training_document_info[0], training_document_info[0]],
properties=custom_form_model_properties[0],
model_name="my model"
)
model_repr = "CustomFormModel(model_id=1, status=creating, training_started_on=0001-01-01 00:00:00, " \
"training_completed_on=0001-01-01 00:00:00, submodels=[{}, {}], errors=[{}], training_documents=[{}, {}], " \
"model_name=my model, properties={})".format(
custom_form_sub_model[1], custom_form_sub_model[1], form_recognizer_error[1], training_document_info[1], training_document_info[1],
custom_form_model_properties[1]
)[:1024]
assert repr(model) == model_repr
def test_custom_form_model_info(self, custom_form_model_properties):
model = _models.CustomFormModelInfo(
model_id=1, status=_models.CustomFormModelStatus.READY, training_started_on=datetime.datetime(1, 1, 1), training_completed_on=datetime.datetime(1, 1, 1),
properties=custom_form_model_properties[0], model_name="my model"
)
model_repr = "CustomFormModelInfo(model_id=1, status=ready, training_started_on=0001-01-01 00:00:00, training_completed_on=0001-01-01 00:00:00, properties={}, model_name=my model)".format(custom_form_model_properties[1])[:1024]
assert repr(model) == model_repr
def test_account_properties(self):
model = _models.AccountProperties(custom_model_count=100, custom_model_limit=1000)
model_repr = "AccountProperties(custom_model_count=100, custom_model_limit=1000)"
assert repr(model) == model_repr
|
