Thanks to visit codestin.com
Credit goes to github.com

Skip to content

Commit ef9b2de

Browse files
author
Daniel Valdivia
committed
Updated test for Token
1 parent d12defa commit ef9b2de

File tree

1 file changed

+65
-7
lines changed

1 file changed

+65
-7
lines changed

‎language/tests/unit/test_syntax.py‎

Lines changed: 65 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -124,15 +124,39 @@ def test_constructor(self):
124124

125125
text_content = 'All'
126126
text_begin = -1
127-
part_of_speech = PartOfSpeech.DETERMINER
127+
aspect = 'ASPECT_UNKNOWN'
128+
reciprocity = 'RECIPROCITY_UNKNOWN'
129+
case = 'NOMINATIVE'
130+
mood = 'MOOD_UNKNOWN'
131+
tag = 'PRON'
132+
person = 'FIRST'
133+
number = 'SINGULAR'
134+
tense = 'TENSE_UNKNOWN'
135+
form = 'FORM_UNKNOWN'
136+
proper = 'PROPER_UNKNOWN'
137+
voice = 'VOICE_UNKNOWN'
138+
gender = 'GENDER_UNKNOWN'
139+
part_of_speech = PartOfSpeech(aspect, reciprocity, case, mood, tag, person,
140+
number, tense, form, proper, voice, gender)
128141
edge_index = 3
129142
edge_label = 'PREDET'
130143
lemma = text_content
131144
token = self._make_one(text_content, text_begin, part_of_speech,
132145
edge_index, edge_label, lemma)
133146
self.assertEqual(token.text_content, text_content)
134147
self.assertEqual(token.text_begin, text_begin)
135-
self.assertEqual(token.part_of_speech, part_of_speech)
148+
self.assertEqual(token.part_of_speech.aspect, part_of_speech.aspect)
149+
self.assertEqual(token.part_of_speech.reciprocity, part_of_speech.reciprocity)
150+
self.assertEqual(token.part_of_speech.case, part_of_speech.case)
151+
self.assertEqual(token.part_of_speech.mood, part_of_speech.mood)
152+
self.assertEqual(token.part_of_speech.tag, part_of_speech.tag)
153+
self.assertEqual(token.part_of_speech.person, part_of_speech.person)
154+
self.assertEqual(token.part_of_speech.number, part_of_speech.number)
155+
self.assertEqual(token.part_of_speech.tense, part_of_speech.tense)
156+
self.assertEqual(token.part_of_speech.form, part_of_speech.form)
157+
self.assertEqual(token.part_of_speech.proper, part_of_speech.proper)
158+
self.assertEqual(token.part_of_speech.voice, part_of_speech.voice)
159+
self.assertEqual(token.part_of_speech.gender, part_of_speech.gender)
136160
self.assertEqual(token.edge_index, edge_index)
137161
self.assertEqual(token.edge_label, edge_label)
138162
self.assertEqual(token.lemma, lemma)
@@ -143,7 +167,32 @@ def test_from_api_repr(self):
143167
klass = self._get_target_class()
144168
text_content = 'pretty'
145169
text_begin = -1
146-
part_of_speech = PartOfSpeech.ADJECTIVE
170+
aspect = 'ASPECT_UNKNOWN'
171+
reciprocity = 'RECIPROCITY_UNKNOWN'
172+
case = 'NOMINATIVE'
173+
mood = 'MOOD_UNKNOWN'
174+
tag = 'PRON'
175+
person = 'FIRST'
176+
number = 'SINGULAR'
177+
tense = 'TENSE_UNKNOWN'
178+
form = 'FORM_UNKNOWN'
179+
proper = 'PROPER_UNKNOWN'
180+
voice = 'VOICE_UNKNOWN'
181+
gender = 'GENDER_UNKNOWN'
182+
part_of_speech = {
183+
'aspect': aspect,
184+
'reciprocity': reciprocity,
185+
'case': case,
186+
'mood': mood,
187+
'tag': tag,
188+
'person': person,
189+
'number': number,
190+
'tense': tense,
191+
'form': form,
192+
'proper': proper,
193+
'voice': voice,
194+
'gender': gender
195+
}
147196
edge_index = 3
148197
edge_label = 'AMOD'
149198
lemma = text_content
@@ -152,9 +201,7 @@ def test_from_api_repr(self):
152201
'content': text_content,
153202
'beginOffset': text_begin,
154203
},
155-
'partOfSpeech': {
156-
'tag': part_of_speech,
157-
},
204+
'partOfSpeech': part_of_speech,
158205
'dependencyEdge': {
159206
'headTokenIndex': edge_index,
160207
'label': edge_label,
@@ -164,7 +211,18 @@ def test_from_api_repr(self):
164211
token = klass.from_api_repr(payload)
165212
self.assertEqual(token.text_content, text_content)
166213
self.assertEqual(token.text_begin, text_begin)
167-
self.assertEqual(token.part_of_speech, part_of_speech)
214+
self.assertEqual(token.part_of_speech.aspect, part_of_speech['aspect'])
215+
self.assertEqual(token.part_of_speech.reciprocity, part_of_speech['reciprocity'])
216+
self.assertEqual(token.part_of_speech.case, part_of_speech['case'])
217+
self.assertEqual(token.part_of_speech.mood, part_of_speech['mood'])
218+
self.assertEqual(token.part_of_speech.tag, part_of_speech['tag'])
219+
self.assertEqual(token.part_of_speech.person, part_of_speech['person'])
220+
self.assertEqual(token.part_of_speech.number, part_of_speech['number'])
221+
self.assertEqual(token.part_of_speech.tense, part_of_speech['tense'])
222+
self.assertEqual(token.part_of_speech.form, part_of_speech['form'])
223+
self.assertEqual(token.part_of_speech.proper, part_of_speech['proper'])
224+
self.assertEqual(token.part_of_speech.voice, part_of_speech['voice'])
225+
self.assertEqual(token.part_of_speech.gender, part_of_speech['gender'])
168226
self.assertEqual(token.edge_index, edge_index)
169227
self.assertEqual(token.edge_label, edge_label)
170228
self.assertEqual(token.lemma, lemma)

0 commit comments

Comments
 (0)