o hf# @sdZddlZddlZddlmmZddlmZddlm Z ddl m Z ddl Tej jej jgZddd d d d dd dd dddddddddd ddddddddddd ddddd dd!d"d#ggZddd d d d dd dd dddddddddd dddddddddd ddddd dd!d"d#gddd d d d dd d$d ddddddddd%d ddddddddd&d ddddd dd!d'd#ggZeed(Zeed)ZdCd*d+Zd,d-Zd.d/Zd0d1Zd2d3Zd4d5Zd6d7Zd8d9Zd:d;Zdd?Zd@ Z!dAdBZ"dS)Dz Test the semgrex interface N)Document)SemgrexRequest)CoNLL)*UnbanunbanVERBVBzMood=Imp|VerbForm=Finrootzstart_char=0|end_char=5) idtextlemmauposxposfeatsheaddeprelmiscMoxPROPNNNPz Number=Singcompoundzstart_char=6|end_char=9Opalobjzstart_char=10|end_char=14GEM) r r rrrrrrrner!PUNCT.punctzstart_char=14|end_char=15)r r rrrrrrzstart_char=16|end_char=21zstart_char=22|end_char=25zstart_char=26|end_char=30zstart_char=30|end_char=31zUnban Mox Opal!zUnban Mox Opal! Unban Mox Opal!cCst|j|ks Jt|jdj|ksJ|jdjD]i}t|jdks&J|jdj|ks0J|jD]P}t|jdks>J|jdjdksHJ|jdj|ksRJ|jdjdks\J|jdj|ksfJt|jdksoJ|jdjdksyJ|jdj|ksJq3qdS)Nrrrsourcetargetzzz)lenresultmatch matchIndexnodenamereln)response response_len semgrex_len source_index target_indexr-semgrex_resultr)r4[/var/www/html/env_mimamsha/lib/python3.10/site-packages/stanza/tests/server/test_semgrex.pycheck_responses  r6cCsnt)}|td}t||td}t||td}t|ddWddS1s0wYdSN{}=source >obj=zzz {}=targetr)r/)semgrexSemgrexprocessONE_SENTENCE_DOCr6TWO_SENTENCE_DOC)semr.r4r4r5 test_multis    "r?cCsttd}t|dS)Nr8r9 process_docr<r6r.r4r4r5test_single_sentences  rCcCsttdd}t|dddS)Nr8r)r0r@rBr4r4r5test_two_semgrexsrDcCsttd}t|dddSr7)r9rAr=r6rBr4r4r5test_two_sentencess rEcC ttd}t|dddddS)Nz*{word:Mox}=source <=zzz {word:Opal}=targetrrrr/r1r-r@rBr4r4r5test_word_attribute rHcCrF)Nz,{lemma:Mox}=source <=zzz {lemma:Opal}=targetrrrrGr@rBr4r4r5test_lemma_attributerIrJcCs<ttd}t|ddddttd}t|dddddS)Nz){tag:NNP}=source <=zzz {word:Opal}=targetrrrrGz){pos:NNP}=source <=zzz {word:Opal}=targetr@rBr4r4r5test_xpos_attributes  rKcCrF)Nz,{cpos:PROPN}=source <=zzz {word:Opal}=targetrrrrGr@rBr4r4r5test_upos_attributerIrLcCrF)Nz*{cpos:PROPN}=source <=zzz {ner:GEM}=targetrrrrGr@rBr4r4r5test_ner_attributerIrMcCst}|jd|j}tgdD]\}}|j}||_||_|j j }d|_ |d|_ q|j j }d|_d|_d|_|j j }d|_d|_d|_t|}t|dS) z} Essentially a test program: the result should be a response with one match, two named nodes, one named relation r8)rrrrrrrrN)rr9appendqueryadd enumeratetokenwordvaluegraphr+ sentenceIndexindexedger$r%depsend_semgrex_requestr6)requestrOidxrSrRr+rXr.r4r4r5test_hand_built_requests(         r]a # sent_id = weblog-juancole.com_juancole_20051126063000_ENG_20051126_063000-0007 # text = You wonder if he was manipulating the market with his bombing targets. 1 You you PRON PRP Case=Nom|Person=2|PronType=Prs 2 nsubj _ _ 2 wonder wonder VERB VBP Mood=Ind|Number=Sing|Person=2|Tense=Pres|VerbForm=Fin 1 _ _ _ 3 if if SCONJ IN _ 6 mark _ _ 4 he he PRON PRP Case=Nom|Gender=Masc|Number=Sing|Person=3|PronType=Prs 6 nsubj _ _ 5 was be AUX VBD Mood=Ind|Number=Sing|Person=3|Tense=Past|VerbForm=Fin 6 aux _ _ 6 manipulating manipulate VERB VBG Tense=Pres|VerbForm=Part 2 ccomp _ _ 7 the the DET DT Definite=Def|PronType=Art 8 det _ _ 8 market market NOUN NN Number=Sing 6 obj _ _ 9 with with ADP IN _ 12 case _ _ 10 his his PRON PRP$ Case=Gen|Gender=Masc|Number=Sing|Person=3|Poss=Yes|PronType=Prs 12 nmod:poss _ _ 11 bombing bombing NOUN NN Number=Sing 12 compound _ _ 12 targets target NOUN NNS Number=Plur 6 obl _ SpaceAfter=No 13 . . PUNCT . _ 2 punct _ _ cCsttjtd}t|d}t|}t|jdksJt|jdjdks&Jt|jdjdjdks5Jt|jdjdjdj dksGJ|jdjdjdj dj dksZJ|jdjdjdj dj dksmJt|jdjdjdj dksJ|jdjdjdj dj dksJ|jdjdjdj djdksJ|jdjdjdj djdksJdS) zg A user / contributor sent a dependency file with blank dependency labels and twisted up roots ) input_strz{}=root <_=edge {}rrr r_N)r conll2docBLANK_DEPENDENCY_SENTENCEr9 build_requestrZr'r(r)r+r,r*rXr$r%r-) blank_dep_docblank_dep_requestr.r4r4r5test_blank_dependencys   $&&$&&*re)rrrrr)#__doc__pyteststanzastanza.server.semgrexserverr9stanza.models.common.docrstanza.protobufrstanza.utils.conllr stanza.testsmarktravisclient pytestmarkTEST_ONE_SENTENCETEST_TWO_SENTENCESr<r=r6r?rCrDrErHrJrKrLrMr]lstriprarer4r4r4r5s(      0   -    X