@@ -328,8 +328,10 @@ def test_is_split_into_words(self):
328
328
self .assertEqual (
329
329
nested_simplify (output ),
330
330
[
331
- {"entity_group" : "PER" , "score" : ANY (float ), "word" : "Sarah" , "start" : 6 , "end" : 11 },
332
- {"entity_group" : "LOC" , "score" : ANY (float ), "word" : "New York" , "start" : 21 , "end" : 29 },
331
+ [
332
+ {"entity_group" : "PER" , "score" : ANY (float ), "word" : "Sarah" , "start" : 6 , "end" : 11 },
333
+ {"entity_group" : "LOC" , "score" : ANY (float ), "word" : "New York" , "start" : 21 , "end" : 29 },
334
+ ]
333
335
],
334
336
)
335
337
@@ -349,8 +351,8 @@ def test_is_split_into_words(self):
349
351
{"entity_group" : "LOC" , "score" : ANY (float ), "word" : "New York" , "start" : 21 , "end" : 29 },
350
352
],
351
353
[
352
- {"entity_group" : "PER" , "score" : ANY (float ), "word" : "Wolfgang" , "start" : 12 , "end" : 20 },
353
- {"entity_group" : "LOC" , "score" : ANY (float ), "word" : "Berlin" , "start" : 36 , "end" : 42 },
354
+ {"entity_group" : "PER" , "score" : ANY (float ), "word" : "Wolfgang" , "start" : 11 , "end" : 19 },
355
+ {"entity_group" : "LOC" , "score" : ANY (float ), "word" : "Berlin" , "start" : 34 , "end" : 40 },
354
356
],
355
357
],
356
358
)
0 commit comments