B
    .(äb‰  ã               @   sH   d dl mZ d dlmZ d dlmZ d dlmZ d dlm	Z	 dd„ Z
dS )	é    )Úunicode_literals)Údefaultdict)ÚEntityRecognizer)ÚEnglish)ÚSpanc              C   s  d} d}t ƒ }t|jƒ}| d¡ | g ¡ | |¡ |dƒ}t|jƒdksPt‚d|jks^t‚t	|dddd	}t
|jƒ|g |_|g}|jj|| |d
}xdt||ƒD ]V\}}ttƒ}	xD|jj |¡D ]2\}
}x(|D ] \}}}|	|||f  |
7  < qÒW qÄW q¤W dS )z< This should not crash or exit with some strange error code é   g-Cëâ6?Z
SOME_LABELzWhat do you think about Apple ?é   é   é   ZMY_ORG)Úlabel)Ú
beam_widthÚbeam_densityN)r   r   ZvocabZ	add_labelZbegin_trainingZadd_pipeÚlenÚlabelsÚAssertionErrorr   ÚlistÚentsÚentityZ
beam_parseÚzipr   ÚfloatÚmovesZget_beam_parses)r   r   ZnlpZnerÚdocZ	apple_entZdocsZbeamsZbeamZentity_scoresZscorer   ÚstartÚendr   © r   úT/home/dcms/DCMS/lib/python3.7/site-packages/spacy/tests/regression/test_issue4313.pyÚtest_issue4313   s(    



r   N)Ú
__future__r   Úcollectionsr   Zspacy.pipeliner   Zspacy.lang.enr   Zspacy.tokensr   r   r   r   r   r   Ú<module>   s
   