B .(b @shddlmZddlZddlmZmZddlmZddlm Z ddl m Z ddl m Z d d Zd d ZdS) )unicode_literalsN)EntityRecognizer EntityRuler)English)Span) ensure_path) make_tempdirc Cst}|d}|d|||t|}ddddddidd igdg}|||j|dd |d }|jd jdkst t N}t |}| s| ||t|}|d }|jd jdkst Wd QRXd S)z@Test that serialization of an EntityRuler before NER works fine.ner SOME_LABELMY_ORGZApple)labelpatternZMY_GPElowersanZ francisco)beforezWhat do you think about Apple ?rN)r create_pipe add_labeladd_pipebegin_trainingrZ add_patternsentsZlabel_AssertionErrorr rexistsmkdirto_diskspacyload) Znlpr Zrulerpatternsdoc1d output_dirZnlp2Zdoc2r!T/home/dcms/DCMS/lib/python3.7/site-packages/spacy/tests/regression/test_issue4042.pytest_issue4042 s(      r#c Cst}|j}|d}|d||||d}t|jdksLtd|jksZtt |dddd}t |j |g|_ ||t|jd kstd|jkstd|jkstt N}t |}|s|||t|}||t|jd kstWd QRXd S) z Test that serialization of an NER works fine when new labels were added. This is the second bug of two bugs underlying the issue 4042. r r zWhat do you think about Apple ?r )r rN)rvocabrrrrlenlabelsrrlistrr rrrrrZ from_disk)Znlp1r'Zner1rZ apple_entrr Zner2r!r!r"test_issue4042_bug2-s.     r+) __future__rrZspacy.pipelinerrZ spacy.lang.enrZ spacy.tokensrZ spacy.utilrutilr r#r+r!r!r!r"s