nltk 3.2__zip → 3.3__zip

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (406) hide show
  1. {nltk-3.2 → nltk-3.3}/INSTALL.txt +6 -6
  2. {nltk-3.2 → nltk-3.3}/LICENSE.txt +13 -13
  3. {nltk-3.2 → nltk-3.3}/MANIFEST.in +6 -6
  4. {nltk-3.2/nltk.egg-info → nltk-3.3}/PKG-INFO +13 -5
  5. nltk-3.2/README.txt → nltk-3.3/README.md +49 -48
  6. nltk-3.3/nltk/VERSION +1 -0
  7. {nltk-3.2 → nltk-3.3}/nltk/__init__.py +185 -185
  8. {nltk-3.2 → nltk-3.3}/nltk/app/__init__.py +52 -53
  9. {nltk-3.2 → nltk-3.3}/nltk/app/chartparser_app.py +2276 -2276
  10. {nltk-3.2 → nltk-3.3}/nltk/app/chunkparser_app.py +1262 -1264
  11. {nltk-3.2 → nltk-3.3}/nltk/app/collocations_app.py +347 -350
  12. {nltk-3.2 → nltk-3.3}/nltk/app/concordance_app.py +568 -570
  13. {nltk-3.2 → nltk-3.3}/nltk/app/nemo_app.py +156 -155
  14. {nltk-3.2 → nltk-3.3}/nltk/app/rdparser_app.py +892 -894
  15. {nltk-3.2 → nltk-3.3}/nltk/app/srparser_app.py +809 -809
  16. {nltk-3.2 → nltk-3.3}/nltk/app/wordfreq_app.py +32 -32
  17. {nltk-3.2 → nltk-3.3}/nltk/app/wordnet_app.py +968 -968
  18. {nltk-3.2 → nltk-3.3}/nltk/book.py +91 -90
  19. {nltk-3.2 → nltk-3.3}/nltk/ccg/__init__.py +22 -22
  20. {nltk-3.2 → nltk-3.3}/nltk/ccg/api.py +359 -335
  21. {nltk-3.2 → nltk-3.3}/nltk/ccg/chart.py +394 -361
  22. {nltk-3.2 → nltk-3.3}/nltk/ccg/combinator.py +343 -312
  23. {nltk-3.2 → nltk-3.3}/nltk/ccg/lexicon.py +328 -290
  24. nltk-3.3/nltk/ccg/logic.py +46 -0
  25. {nltk-3.2 → nltk-3.3}/nltk/chat/__init__.py +49 -49
  26. {nltk-3.2 → nltk-3.3}/nltk/chat/eliza.py +244 -244
  27. {nltk-3.2 → nltk-3.3}/nltk/chat/iesha.py +140 -140
  28. {nltk-3.2 → nltk-3.3}/nltk/chat/rude.py +92 -92
  29. {nltk-3.2 → nltk-3.3}/nltk/chat/suntsu.py +117 -117
  30. {nltk-3.2 → nltk-3.3}/nltk/chat/util.py +122 -120
  31. {nltk-3.2 → nltk-3.3}/nltk/chat/zen.py +282 -282
  32. {nltk-3.2 → nltk-3.3}/nltk/chunk/__init__.py +190 -190
  33. {nltk-3.2 → nltk-3.3}/nltk/chunk/api.py +51 -51
  34. {nltk-3.2 → nltk-3.3}/nltk/chunk/named_entity.py +331 -331
  35. {nltk-3.2 → nltk-3.3}/nltk/chunk/regexp.py +1390 -1384
  36. {nltk-3.2 → nltk-3.3}/nltk/chunk/util.py +600 -600
  37. {nltk-3.2 → nltk-3.3}/nltk/classify/__init__.py +98 -98
  38. {nltk-3.2 → nltk-3.3}/nltk/classify/api.py +193 -193
  39. {nltk-3.2 → nltk-3.3}/nltk/classify/decisiontree.py +295 -295
  40. {nltk-3.2 → nltk-3.3}/nltk/classify/maxent.py +1510 -1492
  41. {nltk-3.2 → nltk-3.3}/nltk/classify/megam.py +179 -178
  42. {nltk-3.2 → nltk-3.3}/nltk/classify/naivebayes.py +245 -242
  43. {nltk-3.2 → nltk-3.3}/nltk/classify/positivenaivebayes.py +170 -170
  44. {nltk-3.2 → nltk-3.3}/nltk/classify/rte_classify.py +160 -191
  45. {nltk-3.2 → nltk-3.3}/nltk/classify/scikitlearn.py +153 -151
  46. {nltk-3.2 → nltk-3.3}/nltk/classify/senna.py +183 -181
  47. {nltk-3.2 → nltk-3.3}/nltk/classify/svm.py +15 -15
  48. {nltk-3.2 → nltk-3.3}/nltk/classify/tadm.py +113 -112
  49. {nltk-3.2 → nltk-3.3}/nltk/classify/textcat.py +193 -193
  50. {nltk-3.2 → nltk-3.3}/nltk/classify/util.py +324 -312
  51. {nltk-3.2 → nltk-3.3}/nltk/classify/weka.py +346 -345
  52. {nltk-3.2 → nltk-3.3}/nltk/cluster/__init__.py +86 -86
  53. {nltk-3.2 → nltk-3.3}/nltk/cluster/api.py +74 -70
  54. {nltk-3.2 → nltk-3.3}/nltk/cluster/em.py +247 -247
  55. {nltk-3.2 → nltk-3.3}/nltk/cluster/gaac.py +168 -168
  56. {nltk-3.2 → nltk-3.3}/nltk/cluster/kmeans.py +221 -221
  57. {nltk-3.2 → nltk-3.3}/nltk/cluster/util.py +304 -289
  58. nltk-3.3/nltk/collections.py +640 -0
  59. {nltk-3.2 → nltk-3.3}/nltk/collocations.py +388 -388
  60. nltk-3.3/nltk/compat.py +365 -0
  61. {nltk-3.2 → nltk-3.3}/nltk/corpus/__init__.py +341 -320
  62. {nltk-3.2 → nltk-3.3}/nltk/corpus/europarl_raw.py +44 -44
  63. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/__init__.py +147 -145
  64. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/aligned.py +115 -114
  65. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/api.py +448 -446
  66. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/bnc.py +252 -252
  67. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/bracket_parse.py +227 -227
  68. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/categorized_sents.py +179 -178
  69. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/chasen.py +140 -139
  70. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/childes.py +522 -494
  71. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/chunked.py +212 -212
  72. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/cmudict.py +97 -95
  73. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/comparative_sents.py +280 -278
  74. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/conll.py +528 -521
  75. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/crubadan.py +116 -116
  76. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/dependency.py +101 -101
  77. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/framenet.py +3059 -2093
  78. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/ieer.py +112 -111
  79. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/indian.py +85 -86
  80. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/ipipan.py +332 -331
  81. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/knbc.py +187 -187
  82. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/lin.py +156 -156
  83. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/mte.py +311 -310
  84. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/nkjp.py +429 -428
  85. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/nombank.py +421 -421
  86. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/nps_chat.py +73 -73
  87. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/opinion_lexicon.py +115 -115
  88. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/panlex_lite.py +165 -165
  89. nltk-3.3/nltk/corpus/reader/pl196x.py +292 -0
  90. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/plaintext.py +232 -233
  91. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/ppattach.py +96 -95
  92. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/propbank.py +479 -481
  93. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/pros_cons.py +128 -126
  94. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/reviews.py +330 -328
  95. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/rte.py +144 -146
  96. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/semcor.py +256 -256
  97. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/senseval.py +203 -201
  98. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/sentiwordnet.py +137 -137
  99. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/sinica_treebank.py +75 -75
  100. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/string_category.py +60 -59
  101. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/switchboard.py +119 -119
  102. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/tagged.py +295 -294
  103. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/timit.py +452 -450
  104. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/toolbox.py +68 -68
  105. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/twitter.py +155 -157
  106. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/udhr.py +80 -80
  107. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/util.py +804 -800
  108. nltk-3.3/nltk/corpus/reader/verbnet.py +595 -0
  109. nltk-3.3/nltk/corpus/reader/wordlist.py +134 -0
  110. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/wordnet.py +2108 -2036
  111. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/xmldocs.py +392 -387
  112. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/ycoe.py +243 -242
  113. {nltk-3.2 → nltk-3.3}/nltk/corpus/util.py +144 -127
  114. {nltk-3.2 → nltk-3.3}/nltk/data.py +1487 -1451
  115. {nltk-3.2 → nltk-3.3}/nltk/decorators.py +219 -215
  116. {nltk-3.2 → nltk-3.3}/nltk/downloader.py +2278 -2274
  117. {nltk-3.2 → nltk-3.3}/nltk/draw/__init__.py +27 -27
  118. {nltk-3.2 → nltk-3.3}/nltk/draw/cfg.py +774 -777
  119. {nltk-3.2 → nltk-3.3}/nltk/draw/dispersion.py +58 -58
  120. {nltk-3.2 → nltk-3.3}/nltk/draw/table.py +1098 -1098
  121. {nltk-3.2 → nltk-3.3}/nltk/draw/tree.py +963 -966
  122. {nltk-3.2 → nltk-3.3}/nltk/draw/util.py +2356 -2356
  123. {nltk-3.2 → nltk-3.3}/nltk/featstruct.py +2617 -2500
  124. {nltk-3.2 → nltk-3.3}/nltk/grammar.py +1559 -1516
  125. {nltk-3.2 → nltk-3.3}/nltk/help.py +56 -56
  126. {nltk-3.2 → nltk-3.3}/nltk/inference/__init__.py +20 -20
  127. {nltk-3.2 → nltk-3.3}/nltk/inference/api.py +597 -589
  128. {nltk-3.2 → nltk-3.3}/nltk/inference/discourse.py +612 -608
  129. {nltk-3.2 → nltk-3.3}/nltk/inference/mace.py +311 -311
  130. {nltk-3.2 → nltk-3.3}/nltk/inference/nonmonotonic.py +509 -509
  131. {nltk-3.2 → nltk-3.3}/nltk/inference/prover9.py +431 -431
  132. {nltk-3.2 → nltk-3.3}/nltk/inference/resolution.py +687 -687
  133. {nltk-3.2 → nltk-3.3}/nltk/inference/tableau.py +607 -607
  134. {nltk-3.2 → nltk-3.3}/nltk/internals.py +982 -982
  135. {nltk-3.2 → nltk-3.3}/nltk/jsontags.py +63 -63
  136. {nltk-3.2 → nltk-3.3}/nltk/lazyimport.py +142 -142
  137. {nltk-3.2 → nltk-3.3}/nltk/metrics/__init__.py +30 -29
  138. {nltk-3.2 → nltk-3.3}/nltk/metrics/agreement.py +421 -401
  139. nltk-3.3/nltk/metrics/aline.py +607 -0
  140. {nltk-3.2 → nltk-3.3}/nltk/metrics/association.py +414 -410
  141. {nltk-3.2 → nltk-3.3}/nltk/metrics/confusionmatrix.py +206 -206
  142. {nltk-3.2 → nltk-3.3}/nltk/metrics/distance.py +207 -202
  143. {nltk-3.2 → nltk-3.3}/nltk/metrics/paice.py +381 -381
  144. {nltk-3.2 → nltk-3.3}/nltk/metrics/scores.py +228 -228
  145. {nltk-3.2 → nltk-3.3}/nltk/metrics/segmentation.py +231 -232
  146. {nltk-3.2 → nltk-3.3}/nltk/metrics/spearman.py +69 -69
  147. {nltk-3.2 → nltk-3.3}/nltk/misc/__init__.py +11 -11
  148. {nltk-3.2 → nltk-3.3}/nltk/misc/babelfish.py +10 -10
  149. {nltk-3.2 → nltk-3.3}/nltk/misc/chomsky.py +135 -133
  150. {nltk-3.2 → nltk-3.3}/nltk/misc/minimalset.py +83 -83
  151. {nltk-3.2 → nltk-3.3}/nltk/misc/sort.py +157 -157
  152. {nltk-3.2 → nltk-3.3}/nltk/misc/wordfinder.py +129 -129
  153. {nltk-3.2 → nltk-3.3}/nltk/parse/__init__.py +81 -80
  154. {nltk-3.2 → nltk-3.3}/nltk/parse/api.py +66 -66
  155. {nltk-3.2 → nltk-3.3}/nltk/parse/bllip.py +282 -282
  156. {nltk-3.2 → nltk-3.3}/nltk/parse/chart.py +1682 -1681
  157. nltk-3.3/nltk/parse/corenlp.py +780 -0
  158. {nltk-3.2 → nltk-3.3}/nltk/parse/dependencygraph.py +760 -758
  159. {nltk-3.2 → nltk-3.3}/nltk/parse/earleychart.py +452 -451
  160. {nltk-3.2 → nltk-3.3}/nltk/parse/evaluate.py +129 -129
  161. {nltk-3.2 → nltk-3.3}/nltk/parse/featurechart.py +580 -578
  162. {nltk-3.2 → nltk-3.3}/nltk/parse/generate.py +87 -76
  163. {nltk-3.2 → nltk-3.3}/nltk/parse/malt.py +356 -356
  164. {nltk-3.2 → nltk-3.3}/nltk/parse/nonprojectivedependencyparser.py +783 -783
  165. {nltk-3.2 → nltk-3.3}/nltk/parse/pchart.py +522 -522
  166. {nltk-3.2 → nltk-3.3}/nltk/parse/projectivedependencyparser.py +579 -578
  167. {nltk-3.2 → nltk-3.3}/nltk/parse/recursivedescent.py +655 -656
  168. {nltk-3.2 → nltk-3.3}/nltk/parse/shiftreduce.py +458 -459
  169. {nltk-3.2 → nltk-3.3}/nltk/parse/stanford.py +431 -406
  170. {nltk-3.2 → nltk-3.3}/nltk/parse/transitionparser.py +774 -772
  171. {nltk-3.2 → nltk-3.3}/nltk/parse/util.py +231 -231
  172. {nltk-3.2 → nltk-3.3}/nltk/parse/viterbi.py +401 -401
  173. {nltk-3.2 → nltk-3.3}/nltk/probability.py +2410 -2355
  174. {nltk-3.2 → nltk-3.3}/nltk/sem/__init__.py +61 -61
  175. {nltk-3.2 → nltk-3.3}/nltk/sem/boxer.py +1261 -1261
  176. {nltk-3.2 → nltk-3.3}/nltk/sem/chat80.py +782 -782
  177. {nltk-3.2 → nltk-3.3}/nltk/sem/cooper_storage.py +118 -118
  178. {nltk-3.2 → nltk-3.3}/nltk/sem/drt.py +1258 -1257
  179. {nltk-3.2 → nltk-3.3}/nltk/sem/drt_glue_demo.py +485 -489
  180. {nltk-3.2 → nltk-3.3}/nltk/sem/evaluate.py +790 -784
  181. {nltk-3.2 → nltk-3.3}/nltk/sem/glue.py +696 -694
  182. {nltk-3.2 → nltk-3.3}/nltk/sem/hole.py +386 -384
  183. {nltk-3.2 → nltk-3.3}/nltk/sem/lfg.py +210 -208
  184. {nltk-3.2 → nltk-3.3}/nltk/sem/linearlogic.py +451 -449
  185. {nltk-3.2 → nltk-3.3}/nltk/sem/logic.py +1910 -1910
  186. {nltk-3.2 → nltk-3.3}/nltk/sem/relextract.py +476 -483
  187. {nltk-3.2 → nltk-3.3}/nltk/sem/skolemize.py +101 -101
  188. {nltk-3.2 → nltk-3.3}/nltk/sem/util.py +249 -249
  189. {nltk-3.2 → nltk-3.3}/nltk/sentiment/__init__.py +14 -14
  190. {nltk-3.2 → nltk-3.3}/nltk/sentiment/sentiment_analyzer.py +229 -229
  191. {nltk-3.2 → nltk-3.3}/nltk/sentiment/util.py +762 -752
  192. {nltk-3.2 → nltk-3.3}/nltk/sentiment/vader.py +443 -445
  193. {nltk-3.2 → nltk-3.3}/nltk/stem/__init__.py +31 -33
  194. {nltk-3.2 → nltk-3.3}/nltk/stem/api.py +28 -25
  195. nltk-3.3/nltk/stem/arlstem.py +369 -0
  196. {nltk-3.2 → nltk-3.3}/nltk/stem/isri.py +345 -345
  197. {nltk-3.2 → nltk-3.3}/nltk/stem/lancaster.py +337 -310
  198. nltk-3.3/nltk/stem/porter.py +710 -0
  199. {nltk-3.2 → nltk-3.3}/nltk/stem/regexp.py +61 -61
  200. {nltk-3.2 → nltk-3.3}/nltk/stem/rslp.py +142 -142
  201. {nltk-3.2 → nltk-3.3}/nltk/stem/snowball.py +4236 -3712
  202. {nltk-3.2 → nltk-3.3}/nltk/stem/util.py +22 -12
  203. {nltk-3.2 → nltk-3.3}/nltk/stem/wordnet.py +51 -51
  204. {nltk-3.2 → nltk-3.3}/nltk/tag/__init__.py +152 -127
  205. {nltk-3.2 → nltk-3.3}/nltk/tag/api.py +86 -81
  206. {nltk-3.2 → nltk-3.3}/nltk/tag/brill.py +424 -426
  207. {nltk-3.2 → nltk-3.3}/nltk/tag/brill_trainer.py +608 -608
  208. {nltk-3.2 → nltk-3.3}/nltk/tag/crf.py +204 -200
  209. {nltk-3.2 → nltk-3.3}/nltk/tag/hmm.py +1277 -1277
  210. {nltk-3.2 → nltk-3.3}/nltk/tag/hunpos.py +132 -132
  211. {nltk-3.2 → nltk-3.3}/nltk/tag/mapping.py +101 -101
  212. {nltk-3.2 → nltk-3.3}/nltk/tag/perceptron.py +329 -314
  213. {nltk-3.2 → nltk-3.3}/nltk/tag/senna.py +147 -145
  214. {nltk-3.2 → nltk-3.3}/nltk/tag/sequential.py +740 -736
  215. {nltk-3.2 → nltk-3.3}/nltk/tag/stanford.py +219 -196
  216. {nltk-3.2 → nltk-3.3}/nltk/tag/tnt.py +595 -604
  217. {nltk-3.2 → nltk-3.3}/nltk/tag/util.py +72 -72
  218. {nltk-3.2 → nltk-3.3}/nltk/tbl/__init__.py +28 -28
  219. {nltk-3.2 → nltk-3.3}/nltk/tbl/api.py +1 -1
  220. {nltk-3.2 → nltk-3.3}/nltk/tbl/demo.py +366 -366
  221. {nltk-3.2 → nltk-3.3}/nltk/tbl/erroranalysis.py +39 -39
  222. {nltk-3.2 → nltk-3.3}/nltk/tbl/feature.py +259 -261
  223. {nltk-3.2 → nltk-3.3}/nltk/tbl/rule.py +321 -321
  224. {nltk-3.2 → nltk-3.3}/nltk/tbl/template.py +303 -308
  225. {nltk-3.2 → nltk-3.3}/nltk/test/__init__.py +18 -18
  226. {nltk-3.2 → nltk-3.3}/nltk/test/all.py +24 -23
  227. {nltk-3.2 → nltk-3.3}/nltk/test/bleu.doctest +14 -14
  228. {nltk-3.2 → nltk-3.3}/nltk/test/bnc.doctest +60 -60
  229. {nltk-3.2 → nltk-3.3}/nltk/test/ccg.doctest +376 -395
  230. nltk-3.3/nltk/test/ccg_semantics.doctest +553 -0
  231. {nltk-3.2 → nltk-3.3}/nltk/test/chat80.doctest +234 -234
  232. {nltk-3.2 → nltk-3.3}/nltk/test/childes.doctest +184 -184
  233. {nltk-3.2 → nltk-3.3}/nltk/test/childes_fixt.py +14 -13
  234. {nltk-3.2 → nltk-3.3}/nltk/test/chunk.doctest +373 -373
  235. {nltk-3.2 → nltk-3.3}/nltk/test/classify.doctest +185 -185
  236. {nltk-3.2 → nltk-3.3}/nltk/test/classify_fixt.py +11 -10
  237. nltk-3.3/nltk/test/collections.doctest +20 -0
  238. {nltk-3.2 → nltk-3.3}/nltk/test/collocations.doctest +276 -276
  239. {nltk-3.2 → nltk-3.3}/nltk/test/compat.doctest +134 -138
  240. {nltk-3.2 → nltk-3.3}/nltk/test/compat_fixt.py +9 -8
  241. nltk-3.3/nltk/test/concordance.doctest +55 -0
  242. {nltk-3.2 → nltk-3.3}/nltk/test/corpus.doctest +2200 -2212
  243. {nltk-3.2 → nltk-3.3}/nltk/test/corpus_fixt.py +3 -3
  244. {nltk-3.2 → nltk-3.3}/nltk/test/crubadan.doctest +65 -65
  245. {nltk-3.2 → nltk-3.3}/nltk/test/data.doctest +379 -379
  246. {nltk-3.2 → nltk-3.3}/nltk/test/dependency.doctest +241 -241
  247. {nltk-3.2 → nltk-3.3}/nltk/test/discourse.doctest +546 -546
  248. {nltk-3.2 → nltk-3.3}/nltk/test/discourse_fixt.py +14 -13
  249. {nltk-3.2 → nltk-3.3}/nltk/test/doctest_nose_plugin.py +159 -156
  250. {nltk-3.2 → nltk-3.3}/nltk/test/drt.doctest +517 -517
  251. {nltk-3.2 → nltk-3.3}/nltk/test/featgram.doctest +607 -607
  252. {nltk-3.2 → nltk-3.3}/nltk/test/featstruct.doctest +1229 -1230
  253. {nltk-3.2 → nltk-3.3}/nltk/test/framenet.doctest +288 -239
  254. {nltk-3.2 → nltk-3.3}/nltk/test/generate.doctest +67 -67
  255. {nltk-3.2 → nltk-3.3}/nltk/test/gensim.doctest +140 -141
  256. {nltk-3.2 → nltk-3.3}/nltk/test/gensim_fixt.py +10 -9
  257. {nltk-3.2 → nltk-3.3}/nltk/test/gluesemantics.doctest +384 -386
  258. {nltk-3.2 → nltk-3.3}/nltk/test/gluesemantics_malt.doctest +68 -68
  259. {nltk-3.2 → nltk-3.3}/nltk/test/gluesemantics_malt_fixt.py +12 -11
  260. {nltk-3.2 → nltk-3.3}/nltk/test/grammar.doctest +48 -48
  261. {nltk-3.2 → nltk-3.3}/nltk/test/grammartestsuites.doctest +109 -109
  262. {nltk-3.2 → nltk-3.3}/nltk/test/index.doctest +100 -100
  263. {nltk-3.2 → nltk-3.3}/nltk/test/inference.doctest +534 -534
  264. {nltk-3.2 → nltk-3.3}/nltk/test/inference_fixt.py +12 -11
  265. {nltk-3.2 → nltk-3.3}/nltk/test/internals.doctest +140 -140
  266. {nltk-3.2 → nltk-3.3}/nltk/test/japanese.doctest +48 -48
  267. {nltk-3.2 → nltk-3.3}/nltk/test/logic.doctest +1098 -1098
  268. {nltk-3.2 → nltk-3.3}/nltk/test/metrics.doctest +270 -270
  269. {nltk-3.2 → nltk-3.3}/nltk/test/misc.doctest +118 -118
  270. {nltk-3.2 → nltk-3.3}/nltk/test/nonmonotonic.doctest +286 -286
  271. {nltk-3.2 → nltk-3.3}/nltk/test/nonmonotonic_fixt.py +12 -11
  272. {nltk-3.2 → nltk-3.3}/nltk/test/paice.doctest +35 -35
  273. {nltk-3.2 → nltk-3.3}/nltk/test/parse.doctest +884 -884
  274. {nltk-3.2 → nltk-3.3}/nltk/test/portuguese_en.doctest +565 -565
  275. {nltk-3.2 → nltk-3.3}/nltk/test/portuguese_en_fixt.py +16 -16
  276. {nltk-3.2 → nltk-3.3}/nltk/test/probability.doctest +304 -304
  277. {nltk-3.2 → nltk-3.3}/nltk/test/probability_fixt.py +13 -12
  278. {nltk-3.2 → nltk-3.3}/nltk/test/propbank.doctest +176 -176
  279. {nltk-3.2 → nltk-3.3}/nltk/test/relextract.doctest +263 -263
  280. {nltk-3.2 → nltk-3.3}/nltk/test/resolution.doctest +221 -221
  281. {nltk-3.2 → nltk-3.3}/nltk/test/runtests.py +86 -64
  282. {nltk-3.2 → nltk-3.3}/nltk/test/segmentation_fixt.py +10 -10
  283. {nltk-3.2 → nltk-3.3}/nltk/test/semantics.doctest +665 -665
  284. {nltk-3.2 → nltk-3.3}/nltk/test/semantics_fixt.py +7 -7
  285. {nltk-3.2 → nltk-3.3}/nltk/test/sentiment.doctest +233 -233
  286. {nltk-3.2 → nltk-3.3}/nltk/test/sentiwordnet.doctest +41 -41
  287. {nltk-3.2 → nltk-3.3}/nltk/test/simple.doctest +85 -85
  288. {nltk-3.2 → nltk-3.3}/nltk/test/stem.doctest +78 -78
  289. {nltk-3.2 → nltk-3.3}/nltk/test/tag.doctest +33 -33
  290. {nltk-3.2 → nltk-3.3}/nltk/test/tokenize.doctest +273 -166
  291. {nltk-3.2 → nltk-3.3}/nltk/test/toolbox.doctest +307 -307
  292. {nltk-3.2 → nltk-3.3}/nltk/test/translate.doctest +242 -242
  293. {nltk-3.2 → nltk-3.3}/nltk/test/translate_fixt.py +3 -3
  294. {nltk-3.2 → nltk-3.3}/nltk/test/tree.doctest +1101 -1101
  295. {nltk-3.2 → nltk-3.3}/nltk/test/treeprettyprinter.doctest +127 -127
  296. {nltk-3.2 → nltk-3.3}/nltk/test/treetransforms.doctest +156 -156
  297. nltk-3.3/nltk/test/unit/test_2x_compat.py +63 -0
  298. nltk-3.3/nltk/test/unit/test_aline.py +52 -0
  299. nltk-3.3/nltk/test/unit/test_chunk.py +56 -0
  300. {nltk-3.2 → nltk-3.3}/nltk/test/unit/test_classify.py +55 -53
  301. {nltk-3.2 → nltk-3.3}/nltk/test/unit/test_collocations.py +88 -79
  302. nltk-3.3/nltk/test/unit/test_concordance.py +98 -0
  303. nltk-3.3/nltk/test/unit/test_corenlp.py +1000 -0
  304. {nltk-3.2 → nltk-3.3}/nltk/test/unit/test_corpora.py +200 -183
  305. {nltk-3.2 → nltk-3.3}/nltk/test/unit/test_corpus_views.py +45 -45
  306. {nltk-3.2 → nltk-3.3}/nltk/test/unit/test_hmm.py +88 -88
  307. {nltk-3.2 → nltk-3.3}/nltk/test/unit/test_json2csv_corpus.py +188 -187
  308. {nltk-3.2 → nltk-3.3}/nltk/test/unit/test_naivebayes.py +26 -26
  309. nltk-3.3/nltk/test/unit/test_rte_classify.py +86 -0
  310. {nltk-3.2 → nltk-3.3}/nltk/test/unit/test_seekable_unicode_stream_reader.py +124 -124
  311. nltk-3.3/nltk/test/unit/test_senna.py +74 -0
  312. nltk-3.3/nltk/test/unit/test_stem.py +121 -0
  313. {nltk-3.2 → nltk-3.3}/nltk/test/unit/test_tag.py +20 -20
  314. {nltk-3.2 → nltk-3.3}/nltk/test/unit/test_tgrep.py +628 -626
  315. nltk-3.3/nltk/test/unit/test_tokenize.py +168 -0
  316. {nltk-3.2 → nltk-3.3}/nltk/test/unit/test_twitter_auth.py +185 -185
  317. nltk-3.3/nltk/test/unit/test_wordnet.py +134 -0
  318. nltk-3.3/nltk/test/unit/translate/test_bleu.py +244 -0
  319. nltk-3.3/nltk/test/unit/translate/test_gdfa.py +57 -0
  320. {nltk-3.2 → nltk-3.3}/nltk/test/unit/translate/test_ibm1.py +75 -75
  321. {nltk-3.2 → nltk-3.3}/nltk/test/unit/translate/test_ibm2.py +87 -87
  322. {nltk-3.2 → nltk-3.3}/nltk/test/unit/translate/test_ibm3.py +105 -105
  323. {nltk-3.2 → nltk-3.3}/nltk/test/unit/translate/test_ibm4.py +126 -126
  324. {nltk-3.2 → nltk-3.3}/nltk/test/unit/translate/test_ibm5.py +166 -166
  325. {nltk-3.2 → nltk-3.3}/nltk/test/unit/translate/test_ibm_model.py +270 -270
  326. nltk-3.3/nltk/test/unit/translate/test_nist.py +37 -0
  327. {nltk-3.2 → nltk-3.3}/nltk/test/unit/translate/test_stack_decoder.py +300 -300
  328. {nltk-3.2 → nltk-3.3}/nltk/test/unit/utils.py +41 -41
  329. {nltk-3.2 → nltk-3.3}/nltk/test/util.doctest +48 -48
  330. {nltk-3.2 → nltk-3.3}/nltk/test/wordnet.doctest +595 -581
  331. {nltk-3.2 → nltk-3.3}/nltk/test/wordnet_fixt.py +7 -6
  332. {nltk-3.2 → nltk-3.3}/nltk/test/wordnet_lch.doctest +53 -53
  333. {nltk-3.2 → nltk-3.3}/nltk/test/wsd.doctest +68 -68
  334. {nltk-3.2 → nltk-3.3}/nltk/text.py +663 -610
  335. {nltk-3.2 → nltk-3.3}/nltk/tgrep.py +938 -937
  336. {nltk-3.2 → nltk-3.3}/nltk/tokenize/__init__.py +130 -107
  337. {nltk-3.2 → nltk-3.3}/nltk/tokenize/api.py +77 -78
  338. {nltk-3.2 → nltk-3.3}/nltk/tokenize/casual.py +343 -342
  339. {nltk-3.2 → nltk-3.3}/nltk/tokenize/mwe.py +120 -120
  340. nltk-3.3/nltk/tokenize/nist.py +173 -0
  341. {nltk-3.2 → nltk-3.3}/nltk/tokenize/punkt.py +1620 -1601
  342. {nltk-3.2 → nltk-3.3}/nltk/tokenize/regexp.py +208 -202
  343. nltk-3.3/nltk/tokenize/repp.py +151 -0
  344. {nltk-3.2 → nltk-3.3}/nltk/tokenize/sexpr.py +140 -142
  345. {nltk-3.2 → nltk-3.3}/nltk/tokenize/simple.py +138 -136
  346. {nltk-3.2 → nltk-3.3}/nltk/tokenize/stanford.py +117 -113
  347. nltk-3.3/nltk/tokenize/stanford_segmenter.py +246 -0
  348. {nltk-3.2 → nltk-3.3}/nltk/tokenize/texttiling.py +458 -460
  349. nltk-3.3/nltk/tokenize/toktok.py +155 -0
  350. nltk-3.3/nltk/tokenize/treebank.py +363 -0
  351. nltk-3.3/nltk/tokenize/util.py +261 -0
  352. {nltk-3.2 → nltk-3.3}/nltk/toolbox.py +499 -497
  353. {nltk-3.2 → nltk-3.3}/nltk/translate/__init__.py +24 -24
  354. {nltk-3.2 → nltk-3.3}/nltk/translate/api.py +331 -321
  355. {nltk-3.2 → nltk-3.3}/nltk/translate/bleu_score.py +615 -556
  356. nltk-3.3/nltk/translate/chrf_score.py +137 -0
  357. {nltk-3.2 → nltk-3.3}/nltk/translate/gale_church.py +248 -235
  358. {nltk-3.2 → nltk-3.3}/nltk/translate/gdfa.py +135 -131
  359. nltk-3.3/nltk/translate/gleu_score.py +193 -0
  360. {nltk-3.2 → nltk-3.3}/nltk/translate/ibm1.py +250 -241
  361. {nltk-3.2 → nltk-3.3}/nltk/translate/ibm2.py +308 -308
  362. {nltk-3.2 → nltk-3.3}/nltk/translate/ibm3.py +337 -337
  363. {nltk-3.2 → nltk-3.3}/nltk/translate/ibm4.py +467 -467
  364. {nltk-3.2 → nltk-3.3}/nltk/translate/ibm5.py +639 -639
  365. {nltk-3.2 → nltk-3.3}/nltk/translate/ibm_model.py +536 -533
  366. {nltk-3.2 → nltk-3.3}/nltk/translate/metrics.py +40 -40
  367. nltk-3.3/nltk/translate/nist_score.py +178 -0
  368. {nltk-3.2 → nltk-3.3}/nltk/translate/phrase_based.py +175 -175
  369. {nltk-3.2 → nltk-3.3}/nltk/translate/ribes_score.py +325 -325
  370. {nltk-3.2 → nltk-3.3}/nltk/translate/stack_decoder.py +499 -495
  371. {nltk-3.2 → nltk-3.3}/nltk/tree.py +1605 -1603
  372. {nltk-3.2 → nltk-3.3}/nltk/treeprettyprinter.py +564 -564
  373. {nltk-3.2 → nltk-3.3}/nltk/treetransforms.py +309 -309
  374. {nltk-3.2 → nltk-3.3}/nltk/twitter/__init__.py +28 -28
  375. {nltk-3.2 → nltk-3.3}/nltk/twitter/api.py +141 -136
  376. {nltk-3.2 → nltk-3.3}/nltk/twitter/common.py +264 -258
  377. {nltk-3.2 → nltk-3.3}/nltk/twitter/twitter_demo.py +288 -288
  378. {nltk-3.2 → nltk-3.3}/nltk/twitter/twitterclient.py +530 -530
  379. {nltk-3.2 → nltk-3.3}/nltk/twitter/util.py +147 -146
  380. nltk-3.3/nltk/util.py +751 -0
  381. {nltk-3.2 → nltk-3.3}/nltk/wsd.py +53 -53
  382. {nltk-3.2 → nltk-3.3/nltk.egg-info}/PKG-INFO +13 -5
  383. {nltk-3.2 → nltk-3.3}/nltk.egg-info/SOURCES.txt +26 -3
  384. nltk-3.3/nltk.egg-info/requires.txt +31 -0
  385. {nltk-3.2 → nltk-3.3}/setup.cfg +0 -1
  386. {nltk-3.2 → nltk-3.3}/setup.py +102 -75
  387. nltk-3.2/nltk/VERSION +0 -1
  388. nltk-3.2/nltk/compat.py +0 -680
  389. nltk-3.2/nltk/corpus/reader/pl196x.py +0 -278
  390. nltk-3.2/nltk/corpus/reader/verbnet.py +0 -408
  391. nltk-3.2/nltk/corpus/reader/wordlist.py +0 -37
  392. nltk-3.2/nltk/sentiment/vader_lexicon.txt +0 -7517
  393. nltk-3.2/nltk/six.py +0 -868
  394. nltk-3.2/nltk/stem/porter.py +0 -692
  395. nltk-3.2/nltk/test/unit/test_2x_compat.py +0 -26
  396. nltk-3.2/nltk/test/unit/test_stem.py +0 -36
  397. nltk-3.2/nltk/test/unit/test_tokenize.py +0 -23
  398. nltk-3.2/nltk/test/unit/translate/test_bleu.py +0 -84
  399. nltk-3.2/nltk/tokenize/treebank.py +0 -124
  400. nltk-3.2/nltk/tokenize/util.py +0 -90
  401. nltk-3.2/nltk/util.py +0 -1364
  402. {nltk-3.2 → nltk-3.3}/nltk/test/unit/__init__.py +0 -0
  403. {nltk-3.2 → nltk-3.3}/nltk/test/unit/translate/__init__.py +0 -0
  404. {nltk-3.2 → nltk-3.3}/nltk.egg-info/dependency_links.txt +0 -0
  405. {nltk-3.2 → nltk-3.3}/nltk.egg-info/not-zip-safe +0 -0
  406. {nltk-3.2 → nltk-3.3}/nltk.egg-info/top_level.txt +0 -0
@@ -1,626 +1,628 @@
1
- #!/usr/bin/env python
2
- # -*- coding: utf-8 -*-
3
- #
4
- # Natural Language Toolkit: TGrep search
5
- #
6
- # Copyright (C) 2001-2015 NLTK Project
7
- # Author: Will Roberts <wildwilhelm@gmail.com>
8
- # URL: <http://nltk.org/>
9
- # For license information, see LICENSE.TXT
10
-
11
- '''
12
- Unit tests for nltk.tgrep.
13
- '''
14
-
15
- from __future__ import absolute_import, print_function, unicode_literals
16
- from nltk.compat import b
17
- from nltk.tree import ParentedTree
18
- from nltk import tgrep
19
- import unittest
20
-
21
- class TestSequenceFunctions(unittest.TestCase):
22
-
23
- '''
24
- Class containing unit tests for nltk.tgrep.
25
- '''
26
-
27
- def test_tokenize_simple(self):
28
- '''
29
- Simple test of tokenization.
30
- '''
31
- tokens = tgrep.tgrep_tokenize('A .. (B !< C . D) | ![<< (E , F) $ G]')
32
- self.assertEqual(tokens,
33
- ['A', '..', '(', 'B', '!', '<', 'C', '.', 'D', ')',
34
- '|', '!', '[', '<<', '(', 'E', ',', 'F', ')', '$',
35
- 'G', ']'])
36
-
37
- def test_tokenize_encoding(self):
38
- '''
39
- Test that tokenization handles bytes and strs the same way.
40
- '''
41
- self.assertEqual(
42
- tgrep.tgrep_tokenize(b('A .. (B !< C . D) | ![<< (E , F) $ G]')),
43
- tgrep.tgrep_tokenize('A .. (B !< C . D) | ![<< (E , F) $ G]'))
44
-
45
- def test_tokenize_link_types(self):
46
- '''
47
- Test tokenization of basic link types.
48
- '''
49
- self.assertEqual(tgrep.tgrep_tokenize('A<B'), ['A', '<', 'B'])
50
- self.assertEqual(tgrep.tgrep_tokenize('A>B'), ['A', '>', 'B'])
51
- self.assertEqual(tgrep.tgrep_tokenize('A<3B'), ['A', '<3', 'B'])
52
- self.assertEqual(tgrep.tgrep_tokenize('A>3B'), ['A', '>3', 'B'])
53
- self.assertEqual(tgrep.tgrep_tokenize('A<,B'), ['A', '<,', 'B'])
54
- self.assertEqual(tgrep.tgrep_tokenize('A>,B'), ['A', '>,', 'B'])
55
- self.assertEqual(tgrep.tgrep_tokenize('A<-3B'), ['A', '<-3', 'B'])
56
- self.assertEqual(tgrep.tgrep_tokenize('A>-3B'), ['A', '>-3', 'B'])
57
- self.assertEqual(tgrep.tgrep_tokenize('A<-B'), ['A', '<-', 'B'])
58
- self.assertEqual(tgrep.tgrep_tokenize('A>-B'), ['A', '>-', 'B'])
59
- self.assertEqual(tgrep.tgrep_tokenize('A<\'B'), ['A', '<\'', 'B'])
60
- self.assertEqual(tgrep.tgrep_tokenize('A>\'B'), ['A', '>\'', 'B'])
61
- self.assertEqual(tgrep.tgrep_tokenize('A<:B'), ['A', '<:', 'B'])
62
- self.assertEqual(tgrep.tgrep_tokenize('A>:B'), ['A', '>:', 'B'])
63
- self.assertEqual(tgrep.tgrep_tokenize('A<<B'), ['A', '<<', 'B'])
64
- self.assertEqual(tgrep.tgrep_tokenize('A>>B'), ['A', '>>', 'B'])
65
- self.assertEqual(tgrep.tgrep_tokenize('A<<,B'), ['A', '<<,', 'B'])
66
- self.assertEqual(tgrep.tgrep_tokenize('A>>,B'), ['A', '>>,', 'B'])
67
- self.assertEqual(tgrep.tgrep_tokenize('A<<\'B'), ['A', '<<\'', 'B'])
68
- self.assertEqual(tgrep.tgrep_tokenize('A>>\'B'), ['A', '>>\'', 'B'])
69
- self.assertEqual(tgrep.tgrep_tokenize('A<<:B'), ['A', '<<:', 'B'])
70
- self.assertEqual(tgrep.tgrep_tokenize('A>>:B'), ['A', '>>:', 'B'])
71
- self.assertEqual(tgrep.tgrep_tokenize('A.B'), ['A', '.', 'B'])
72
- self.assertEqual(tgrep.tgrep_tokenize('A,B'), ['A', ',', 'B'])
73
- self.assertEqual(tgrep.tgrep_tokenize('A..B'), ['A', '..', 'B'])
74
- self.assertEqual(tgrep.tgrep_tokenize('A,,B'), ['A', ',,', 'B'])
75
- self.assertEqual(tgrep.tgrep_tokenize('A$B'), ['A', '$', 'B'])
76
- self.assertEqual(tgrep.tgrep_tokenize('A$.B'), ['A', '$.', 'B'])
77
- self.assertEqual(tgrep.tgrep_tokenize('A$,B'), ['A', '$,', 'B'])
78
- self.assertEqual(tgrep.tgrep_tokenize('A$..B'), ['A', '$..', 'B'])
79
- self.assertEqual(tgrep.tgrep_tokenize('A$,,B'), ['A', '$,,', 'B'])
80
- self.assertEqual(tgrep.tgrep_tokenize('A!<B'), ['A', '!', '<', 'B'])
81
- self.assertEqual(tgrep.tgrep_tokenize('A!>B'), ['A', '!', '>', 'B'])
82
- self.assertEqual(tgrep.tgrep_tokenize('A!<3B'), ['A', '!', '<3', 'B'])
83
- self.assertEqual(tgrep.tgrep_tokenize('A!>3B'), ['A', '!', '>3', 'B'])
84
- self.assertEqual(tgrep.tgrep_tokenize('A!<,B'), ['A', '!', '<,', 'B'])
85
- self.assertEqual(tgrep.tgrep_tokenize('A!>,B'), ['A', '!', '>,', 'B'])
86
- self.assertEqual(tgrep.tgrep_tokenize('A!<-3B'),
87
- ['A', '!', '<-3', 'B'])
88
- self.assertEqual(tgrep.tgrep_tokenize('A!>-3B'),
89
- ['A', '!', '>-3', 'B'])
90
- self.assertEqual(tgrep.tgrep_tokenize('A!<-B'), ['A', '!', '<-', 'B'])
91
- self.assertEqual(tgrep.tgrep_tokenize('A!>-B'), ['A', '!', '>-', 'B'])
92
- self.assertEqual(tgrep.tgrep_tokenize('A!<\'B'),
93
- ['A', '!', '<\'', 'B'])
94
- self.assertEqual(tgrep.tgrep_tokenize('A!>\'B'),
95
- ['A', '!', '>\'', 'B'])
96
- self.assertEqual(tgrep.tgrep_tokenize('A!<:B'), ['A', '!', '<:', 'B'])
97
- self.assertEqual(tgrep.tgrep_tokenize('A!>:B'), ['A', '!', '>:', 'B'])
98
- self.assertEqual(tgrep.tgrep_tokenize('A!<<B'), ['A', '!', '<<', 'B'])
99
- self.assertEqual(tgrep.tgrep_tokenize('A!>>B'), ['A', '!', '>>', 'B'])
100
- self.assertEqual(tgrep.tgrep_tokenize('A!<<,B'),
101
- ['A', '!', '<<,', 'B'])
102
- self.assertEqual(tgrep.tgrep_tokenize('A!>>,B'),
103
- ['A', '!', '>>,', 'B'])
104
- self.assertEqual(tgrep.tgrep_tokenize('A!<<\'B'),
105
- ['A', '!', '<<\'', 'B'])
106
- self.assertEqual(tgrep.tgrep_tokenize('A!>>\'B'),
107
- ['A', '!', '>>\'', 'B'])
108
- self.assertEqual(tgrep.tgrep_tokenize('A!<<:B'),
109
- ['A', '!', '<<:', 'B'])
110
- self.assertEqual(tgrep.tgrep_tokenize('A!>>:B'),
111
- ['A', '!', '>>:', 'B'])
112
- self.assertEqual(tgrep.tgrep_tokenize('A!.B'), ['A', '!', '.', 'B'])
113
- self.assertEqual(tgrep.tgrep_tokenize('A!,B'), ['A', '!', ',', 'B'])
114
- self.assertEqual(tgrep.tgrep_tokenize('A!..B'), ['A', '!', '..', 'B'])
115
- self.assertEqual(tgrep.tgrep_tokenize('A!,,B'), ['A', '!', ',,', 'B'])
116
- self.assertEqual(tgrep.tgrep_tokenize('A!$B'), ['A', '!', '$', 'B'])
117
- self.assertEqual(tgrep.tgrep_tokenize('A!$.B'), ['A', '!', '$.', 'B'])
118
- self.assertEqual(tgrep.tgrep_tokenize('A!$,B'), ['A', '!', '$,', 'B'])
119
- self.assertEqual(tgrep.tgrep_tokenize('A!$..B'),
120
- ['A', '!', '$..', 'B'])
121
- self.assertEqual(tgrep.tgrep_tokenize('A!$,,B'),
122
- ['A', '!', '$,,', 'B'])
123
-
124
- def test_tokenize_examples(self):
125
- '''
126
- Test tokenization of the TGrep2 manual example patterns.
127
- '''
128
- self.assertEqual(tgrep.tgrep_tokenize('NP < PP'),
129
- ['NP', '<', 'PP'])
130
- self.assertEqual(tgrep.tgrep_tokenize('/^NP/'),
131
- ['/^NP/'])
132
- self.assertEqual(tgrep.tgrep_tokenize('NP << PP . VP'),
133
- ['NP', '<<', 'PP', '.', 'VP'])
134
- self.assertEqual(tgrep.tgrep_tokenize('NP << PP | . VP'),
135
- ['NP', '<<', 'PP', '|', '.', 'VP'])
136
- self.assertEqual(tgrep.tgrep_tokenize('NP !<< PP [> NP | >> VP]'),
137
- ['NP', '!', '<<', 'PP', '[', '>', 'NP', '|',
138
- '>>', 'VP', ']'])
139
- self.assertEqual(tgrep.tgrep_tokenize('NP << (PP . VP)'),
140
- ['NP', '<<', '(', 'PP', '.', 'VP', ')'])
141
- self.assertEqual(tgrep.tgrep_tokenize('NP <\' (PP <, (IN < on))'),
142
- ['NP', '<\'', '(', 'PP', '<,', '(', 'IN', '<',
143
- 'on', ')', ')'])
144
- self.assertEqual(tgrep.tgrep_tokenize('S < (A < B) < C'),
145
- ['S', '<', '(', 'A', '<', 'B', ')', '<', 'C'])
146
- self.assertEqual(tgrep.tgrep_tokenize('S < ((A < B) < C)'),
147
- ['S', '<', '(', '(', 'A', '<', 'B', ')',
148
- '<', 'C', ')'])
149
- self.assertEqual(tgrep.tgrep_tokenize('S < (A < B < C)'),
150
- ['S', '<', '(', 'A', '<', 'B', '<', 'C', ')'])
151
- self.assertEqual(tgrep.tgrep_tokenize('A<B&.C'),
152
- ['A', '<', 'B', '&', '.', 'C'])
153
-
154
- def test_tokenize_quoting(self):
155
- '''
156
- Test tokenization of quoting.
157
- '''
158
- self.assertEqual(tgrep.tgrep_tokenize('"A<<:B"<<:"A $.. B"<"A>3B"<C'),
159
- ['"A<<:B"', '<<:', '"A $.. B"', '<', '"A>3B"',
160
- '<', 'C'])
161
-
162
- def test_tokenize_nodenames(self):
163
- '''
164
- Test tokenization of node names.
165
- '''
166
- self.assertEqual(tgrep.tgrep_tokenize('Robert'), ['Robert'])
167
- self.assertEqual(tgrep.tgrep_tokenize('/^[Bb]ob/'), ['/^[Bb]ob/'])
168
- self.assertEqual(tgrep.tgrep_tokenize('*'), ['*'])
169
- self.assertEqual(tgrep.tgrep_tokenize('__'), ['__'])
170
- # test tokenization of NLTK tree position syntax
171
- self.assertEqual(tgrep.tgrep_tokenize('N()'),
172
- ['N(', ')'])
173
- self.assertEqual(tgrep.tgrep_tokenize('N(0,)'),
174
- ['N(', '0', ',', ')'])
175
- self.assertEqual(tgrep.tgrep_tokenize('N(0,0)'),
176
- ['N(', '0', ',', '0', ')'])
177
- self.assertEqual(tgrep.tgrep_tokenize('N(0,0,)'),
178
- ['N(', '0', ',', '0', ',', ')'])
179
-
180
- def test_tokenize_macros(self):
181
- '''
182
- Test tokenization of macro definitions.
183
- '''
184
- self.assertEqual(tgrep.tgrep_tokenize(
185
- '@ NP /^NP/;\n@ NN /^NN/;\n@NP [!< NP | < @NN] !$.. @NN'),
186
- ['@', 'NP', '/^NP/', ';', '@', 'NN', '/^NN/', ';',
187
- '@NP', '[', '!', '<', 'NP', '|', '<', '@NN', ']',
188
- '!', '$..', '@NN'])
189
-
190
- def test_node_simple(self):
191
- '''
192
- Test a simple use of tgrep for finding nodes matching a given
193
- pattern.
194
- '''
195
- tree = ParentedTree.fromstring(
196
- '(S (NP (DT the) (JJ big) (NN dog)) '
197
- '(VP bit) (NP (DT a) (NN cat)))')
198
- self.assertEqual(list(tgrep.tgrep_positions('NN', [tree])),
199
- [[(0,2), (2,1)]])
200
- self.assertEqual(list(tgrep.tgrep_nodes('NN', [tree])),
201
- [[tree[0,2], tree[2,1]]])
202
- self.assertEqual(list(tgrep.tgrep_positions('NN|JJ', [tree])),
203
- [[(0, 1), (0, 2), (2, 1)]])
204
-
205
- def test_node_printing(self):
206
- '''Test that the tgrep print operator ' is properly ignored.'''
207
- tree = ParentedTree.fromstring('(S (n x) (N x))')
208
- self.assertEqual(list(tgrep.tgrep_positions('N', [tree])),
209
- list(tgrep.tgrep_positions('\'N', [tree])))
210
- self.assertEqual(list(tgrep.tgrep_positions('/[Nn]/', [tree])),
211
- list(tgrep.tgrep_positions('\'/[Nn]/', [tree])))
212
-
213
- def test_node_encoding(self):
214
- '''
215
- Test that tgrep search strings handles bytes and strs the same
216
- way.
217
- '''
218
- tree = ParentedTree.fromstring(
219
- '(S (NP (DT the) (JJ big) (NN dog)) '
220
- '(VP bit) (NP (DT a) (NN cat)))')
221
- self.assertEqual(list(tgrep.tgrep_positions(b('NN'), [tree])),
222
- list(tgrep.tgrep_positions('NN', [tree])))
223
- self.assertEqual(list(tgrep.tgrep_nodes(b('NN'), [tree])),
224
- list(tgrep.tgrep_nodes('NN', [tree])))
225
- self.assertEqual(list(tgrep.tgrep_positions(b('NN|JJ'), [tree])),
226
- list(tgrep.tgrep_positions('NN|JJ', [tree])))
227
-
228
- def test_node_nocase(self):
229
- '''
230
- Test selecting nodes using case insensitive node names.
231
- '''
232
- tree = ParentedTree.fromstring('(S (n x) (N x))')
233
- self.assertEqual(list(tgrep.tgrep_positions('"N"', [tree])), [[(1,)]])
234
- self.assertEqual(list(tgrep.tgrep_positions('i@"N"', [tree])), [[(0,), (1,)]])
235
-
236
- def test_node_quoted(self):
237
- '''
238
- Test selecting nodes using quoted node names.
239
- '''
240
- tree = ParentedTree.fromstring('(N ("N" x) (N" x) ("\\" x))')
241
- self.assertEqual(list(tgrep.tgrep_positions('"N"', [tree])), [[()]])
242
- self.assertEqual(list(tgrep.tgrep_positions('"\\"N\\""', [tree])), [[(0,)]])
243
- self.assertEqual(list(tgrep.tgrep_positions('"N\\""', [tree])), [[(1,)]])
244
- self.assertEqual(list(tgrep.tgrep_positions('"\\"\\\\\\""', [tree])), [[(2,)]])
245
-
246
- def test_node_regex(self):
247
- '''
248
- Test regex matching on nodes.
249
- '''
250
- tree = ParentedTree.fromstring('(S (NP-SBJ x) (NP x) (NNP x) (VP x))')
251
- # This is a regular expression that matches any node whose
252
- # name starts with NP, including NP-SBJ:
253
- self.assertEqual(list(tgrep.tgrep_positions('/^NP/', [tree])),
254
- [[(0,), (1,)]])
255
-
256
- def test_node_regex_2(self):
257
- '''
258
- Test regex matching on nodes.
259
- '''
260
- tree = ParentedTree.fromstring('(S (SBJ x) (SBJ1 x) (NP-SBJ x))')
261
- self.assertEqual(list(tgrep.tgrep_positions('/^SBJ/', [tree])),
262
- [[(0,), (1,)]])
263
- # This is a regular expression that matches any node whose
264
- # name includes SBJ, including NP-SBJ:
265
- self.assertEqual(list(tgrep.tgrep_positions('/SBJ/', [tree])),
266
- [[(0,), (1,), (2,)]])
267
-
268
- def test_node_tree_position(self):
269
- '''
270
- Test matching on nodes based on NLTK tree position.
271
- '''
272
- tree = ParentedTree.fromstring('(S (NP-SBJ x) (NP x) (NNP x) (VP x))')
273
- # test all tree positions that are not leaves
274
- leaf_positions = set([tree.leaf_treeposition(x)
275
- for x in range(len(tree.leaves()))])
276
- tree_positions = [x for x in tree.treepositions()
277
- if x not in leaf_positions]
278
- for position in tree_positions:
279
- node_id = 'N{0}'.format(position)
280
- tgrep_positions = list(tgrep.tgrep_positions(node_id, [tree]))
281
- self.assertEqual(len(tgrep_positions[0]), 1)
282
- self.assertEqual(tgrep_positions[0][0], position)
283
-
284
- def test_node_noleaves(self):
285
- '''
286
- Test node name matching with the search_leaves flag set to False.
287
- '''
288
- tree = ParentedTree.fromstring('(S (A (T x)) (B (N x)))')
289
- self.assertEqual(list(tgrep.tgrep_positions('x', [tree])),
290
- [[(0, 0, 0), (1, 0, 0)]])
291
- self.assertEqual(list(tgrep.tgrep_positions('x', [tree], False)),
292
- [[]])
293
-
294
- def tests_rel_dominance(self):
295
- '''
296
- Test matching nodes based on dominance relations.
297
- '''
298
- tree = ParentedTree.fromstring('(S (A (T x)) (B (N x)))')
299
- self.assertEqual(list(tgrep.tgrep_positions('* < T', [tree])),
300
- [[(0,)]])
301
- self.assertEqual(list(tgrep.tgrep_positions('* < T > S', [tree])),
302
- [[(0,)]])
303
- self.assertEqual(list(tgrep.tgrep_positions('* !< T', [tree])),
304
- [[(), (0, 0), (0, 0, 0), (1,), (1, 0), (1, 0, 0)]])
305
- self.assertEqual(list(tgrep.tgrep_positions('* !< T > S', [tree])),
306
- [[(1,)]])
307
- self.assertEqual(list(tgrep.tgrep_positions('* > A', [tree])),
308
- [[(0, 0)]])
309
- self.assertEqual(list(tgrep.tgrep_positions('* > B', [tree])),
310
- [[(1, 0)]])
311
- self.assertEqual(list(tgrep.tgrep_positions('* !> B', [tree])),
312
- [[(), (0,), (0, 0), (0, 0, 0), (1,), (1, 0, 0)]])
313
- self.assertEqual(list(tgrep.tgrep_positions('* !> B >> S', [tree])),
314
- [[(0,), (0, 0), (1,)]])
315
- self.assertEqual(list(tgrep.tgrep_positions('* >> S', [tree])),
316
- [[(0,), (0, 0), (1,), (1, 0)]])
317
- self.assertEqual(list(tgrep.tgrep_positions('* >>, S', [tree])),
318
- [[(0,), (0, 0)]])
319
- self.assertEqual(list(tgrep.tgrep_positions('* >>\' S', [tree])),
320
- [[(1,), (1, 0)]])
321
- # Known issue:
322
- #self.assertEqual(list(tgrep.tgrep_positions('* !>> S', [tree])),
323
- # [[()]])
324
- self.assertEqual(list(tgrep.tgrep_positions('* << T', [tree])),
325
- [[(), (0,)]])
326
- self.assertEqual(list(tgrep.tgrep_positions('* <<\' T', [tree])),
327
- [[(0,)]])
328
- self.assertEqual(list(tgrep.tgrep_positions('* <<1 N', [tree])),
329
- [[(1,)]])
330
- self.assertEqual(list(tgrep.tgrep_positions('* !<< T', [tree])),
331
- [[(0, 0), (0, 0, 0), (1,), (1, 0), (1, 0, 0)]])
332
- tree = ParentedTree.fromstring('(S (A (T x)) (B (T x) (N x )))')
333
- self.assertEqual(list(tgrep.tgrep_positions('* <: T', [tree])),
334
- [[(0,)]])
335
- self.assertEqual(list(tgrep.tgrep_positions('* < T', [tree])),
336
- [[(0,), (1,)]])
337
- self.assertEqual(list(tgrep.tgrep_positions('* !<: T', [tree])),
338
- [[(), (0, 0), (0, 0, 0), (1,), (1, 0), (1, 0, 0),
339
- (1, 1), (1, 1, 0)]])
340
- self.assertEqual(list(tgrep.tgrep_positions('* !<: T > S', [tree])),
341
- [[(1,)]])
342
- tree = ParentedTree.fromstring('(S (T (A x) (B x)) (T (C x)))')
343
- self.assertEqual(list(tgrep.tgrep_positions('* >: T', [tree])),
344
- [[(1, 0)]])
345
- self.assertEqual(list(tgrep.tgrep_positions('* !>: T', [tree])),
346
- [[(), (0,), (0, 0), (0, 0, 0), (0, 1), (0, 1, 0),
347
- (1,), (1, 0, 0)]])
348
- tree = ParentedTree.fromstring('(S (A (B (C (D (E (T x))))))'
349
- ' (A (B (C (D (E (T x))) (N x)))))')
350
- self.assertEqual(list(tgrep.tgrep_positions('* <<: T', [tree])),
351
- [[(0,), (0, 0), (0, 0, 0), (0, 0, 0, 0),
352
- (0, 0, 0, 0, 0), (1, 0, 0, 0), (1, 0, 0, 0, 0)]])
353
- self.assertEqual(list(tgrep.tgrep_positions('* >>: A', [tree])),
354
- [[(0, 0), (0, 0, 0), (0, 0, 0, 0), (0, 0, 0, 0, 0),
355
- (0, 0, 0, 0, 0, 0), (1, 0), (1, 0, 0)]])
356
-
357
- def test_bad_operator(self):
358
- '''
359
- Test error handling of undefined tgrep operators.
360
- '''
361
- tree = ParentedTree.fromstring('(S (A (T x)) (B (N x)))')
362
- self.assertRaises(
363
- tgrep.TgrepException,
364
- list,
365
- tgrep.tgrep_positions('* >>> S', [tree]))
366
-
367
- def test_comments(self):
368
- '''
369
- Test that comments are correctly filtered out of tgrep search
370
- strings.
371
- '''
372
- tree = ParentedTree.fromstring('(S (NN x) (NP x) (NN x))')
373
- search1 = '''
374
- @ NP /^NP/;
375
- @ NN /^NN/;
376
- @NN
377
- '''
378
- self.assertEqual(list(tgrep.tgrep_positions(search1, [tree])),
379
- [[(0,), (2,)]])
380
- search2 = '''
381
- # macros
382
- @ NP /^NP/;
383
- @ NN /^NN/;
384
-
385
- # search string
386
- @NN
387
- '''
388
- self.assertEqual(list(tgrep.tgrep_positions(search2, [tree])),
389
- [[(0,), (2,)]])
390
-
391
- def test_rel_sister_nodes(self):
392
- '''
393
- Test matching sister nodes in a tree.
394
- '''
395
- tree = ParentedTree.fromstring('(S (A x) (B x) (C x))')
396
- self.assertEqual(list(tgrep.tgrep_positions('* $. B', [tree])), [[(0,)]])
397
- self.assertEqual(list(tgrep.tgrep_positions('* $.. B', [tree])), [[(0,)]])
398
- self.assertEqual(list(tgrep.tgrep_positions('* $, B', [tree])), [[(2,)]])
399
- self.assertEqual(list(tgrep.tgrep_positions('* $,, B', [tree])), [[(2,)]])
400
- self.assertEqual(list(tgrep.tgrep_positions('* $ B', [tree])), [[(0,), (2,)]])
401
-
402
- def tests_rel_indexed_children(self):
403
- '''
404
- Test matching nodes based on their index in their parent node.
405
- '''
406
- tree = ParentedTree.fromstring('(S (A x) (B x) (C x))')
407
- self.assertEqual(list(tgrep.tgrep_positions('* >, S', [tree])), [[(0,)]])
408
- self.assertEqual(list(tgrep.tgrep_positions('* >1 S', [tree])), [[(0,)]])
409
- self.assertEqual(list(tgrep.tgrep_positions('* >2 S', [tree])), [[(1,)]])
410
- self.assertEqual(list(tgrep.tgrep_positions('* >3 S', [tree])), [[(2,)]])
411
- self.assertEqual(list(tgrep.tgrep_positions('* >\' S', [tree])), [[(2,)]])
412
- self.assertEqual(list(tgrep.tgrep_positions('* >-1 S', [tree])), [[(2,)]])
413
- self.assertEqual(list(tgrep.tgrep_positions('* >-2 S', [tree])), [[(1,)]])
414
- self.assertEqual(list(tgrep.tgrep_positions('* >-3 S', [tree])), [[(0,)]])
415
- tree = ParentedTree.fromstring(
416
- '(S (D (A x) (B x) (C x)) (E (B x) (C x) (A x)) '
417
- '(F (C x) (A x) (B x)))')
418
- self.assertEqual(list(tgrep.tgrep_positions('* <, A', [tree])), [[(0,)]])
419
- self.assertEqual(list(tgrep.tgrep_positions('* <1 A', [tree])), [[(0,)]])
420
- self.assertEqual(list(tgrep.tgrep_positions('* <2 A', [tree])), [[(2,)]])
421
- self.assertEqual(list(tgrep.tgrep_positions('* <3 A', [tree])), [[(1,)]])
422
- self.assertEqual(list(tgrep.tgrep_positions('* <\' A', [tree])), [[(1,)]])
423
- self.assertEqual(list(tgrep.tgrep_positions('* <-1 A', [tree])), [[(1,)]])
424
- self.assertEqual(list(tgrep.tgrep_positions('* <-2 A', [tree])), [[(2,)]])
425
- self.assertEqual(list(tgrep.tgrep_positions('* <-3 A', [tree])), [[(0,)]])
426
-
427
- def test_rel_precedence(self):
428
- '''
429
- Test matching nodes based on precedence relations.
430
- '''
431
- tree = ParentedTree.fromstring('(S (NP (NP (PP x)) (NP (AP x)))'
432
- ' (VP (AP (X (PP x)) (Y (AP x))))'
433
- ' (NP (RC (NP (AP x)))))')
434
- self.assertEqual(list(tgrep.tgrep_positions('* . X', [tree])),
435
- [[(0,), (0, 1), (0, 1, 0)]])
436
- self.assertEqual(list(tgrep.tgrep_positions('* . Y', [tree])),
437
- [[(1, 0, 0), (1, 0, 0, 0)]])
438
- self.assertEqual(list(tgrep.tgrep_positions('* .. X', [tree])),
439
- [[(0,), (0, 0), (0, 0, 0), (0, 1), (0, 1, 0)]])
440
- self.assertEqual(list(tgrep.tgrep_positions('* .. Y', [tree])),
441
- [[(0,), (0, 0), (0, 0, 0), (0, 1), (0, 1, 0),
442
- (1, 0, 0), (1, 0, 0, 0)]])
443
- self.assertEqual(list(tgrep.tgrep_positions('* , X', [tree])),
444
- [[(1, 0, 1), (1, 0, 1, 0)]])
445
- self.assertEqual(list(tgrep.tgrep_positions('* , Y', [tree])),
446
- [[(2,), (2, 0), (2, 0, 0), (2, 0, 0, 0)]])
447
- self.assertEqual(list(tgrep.tgrep_positions('* ,, X', [tree])),
448
- [[(1, 0, 1), (1, 0, 1, 0), (2,), (2, 0), (2, 0, 0),
449
- (2, 0, 0, 0)]])
450
- self.assertEqual(list(tgrep.tgrep_positions('* ,, Y', [tree])),
451
- [[(2,), (2, 0), (2, 0, 0), (2, 0, 0, 0)]])
452
-
453
- def test_examples(self):
454
- '''
455
- Test the Basic Examples from the TGrep2 manual.
456
- '''
457
- tree = ParentedTree.fromstring('(S (NP (AP x)) (NP (PP x)))')
458
- # This matches any NP node that immediately dominates a PP:
459
- self.assertEqual(list(tgrep.tgrep_positions('NP < PP', [tree])),
460
- [[(1,)]])
461
-
462
- tree = ParentedTree.fromstring('(S (NP x) (VP x) (NP (PP x)) (VP x))')
463
- # This matches an NP that dominates a PP and is immediately
464
- # followed by a VP:
465
- self.assertEqual(list(tgrep.tgrep_positions('NP << PP . VP', [tree])),
466
- [[(2,)]])
467
-
468
- tree = ParentedTree.fromstring('(S (NP (AP x)) (NP (PP x)) '
469
- '(NP (DET x) (NN x)) (VP x))')
470
- # This matches an NP that dominates a PP or is immediately
471
- # followed by a VP:
472
- self.assertEqual(list(tgrep.tgrep_positions('NP << PP | . VP', [tree])),
473
- [[(1,), (2,)]])
474
-
475
- tree = ParentedTree.fromstring('(S (NP (NP (PP x)) (NP (AP x)))'
476
- ' (VP (AP (NP (PP x)) (NP (AP x))))'
477
- ' (NP (RC (NP (AP x)))))')
478
- # This matches an NP that does not dominate a PP. Also, the NP
479
- # must either have a parent that is an NP or be dominated by a
480
- # VP:
481
- self.assertEqual(list(tgrep.tgrep_positions(
482
- 'NP !<< PP [> NP | >> VP]', [tree])),
483
- [[(0, 1), (1, 0, 1)]])
484
-
485
- tree = ParentedTree.fromstring('(S (NP (AP (PP x) (VP x))) '
486
- '(NP (AP (PP x) (NP x))) (NP x))')
487
- # This matches an NP that dominates a PP which itself is
488
- # immediately followed by a VP. Note the use of parentheses to
489
- # group ". VP" with the PP rather than with the NP:
490
- self.assertEqual(list(tgrep.tgrep_positions('NP << (PP . VP)', [tree])),
491
- [[(0,)]])
492
-
493
- tree = ParentedTree.fromstring(
494
- '(S (NP (DET a) (NN cat) (PP (IN on) (NP x)))'
495
- ' (NP (DET a) (NN cat) (PP (IN on) (NP x)) (PP x))'
496
- ' (NP x))')
497
- # This matches an NP whose last child is a PP that begins with
498
- # the preposition "on":
499
- self.assertEqual(list(tgrep.tgrep_positions(
500
- 'NP <\' (PP <, (IN < on))', [tree])),
501
- [[(0,)]])
502
-
503
- tree = ParentedTree.fromstring(
504
- '(S (S (C x) (A (B x))) (S (C x) (A x)) '
505
- '(S (D x) (A (B x))))')
506
- # The following pattern matches an S which has a child A and
507
- # another child that is a C and that the A has a child B:
508
- self.assertEqual(list(tgrep.tgrep_positions('S < (A < B) < C', [tree])),
509
- [[(0,)]])
510
-
511
- tree = ParentedTree.fromstring(
512
- '(S (S (A (B x) (C x))) (S (S (C x) (A (B x)))))')
513
- # However, this pattern means that S has child A and that A
514
- # has children B and C:
515
- self.assertEqual(list(tgrep.tgrep_positions('S < ((A < B) < C)', [tree])),
516
- [[(0,)]])
517
-
518
- # It is equivalent to this:
519
- self.assertEqual(list(tgrep.tgrep_positions('S < (A < B < C)', [tree])),
520
- [[(0,)]])
521
-
522
- def test_use_macros(self):
523
- '''
524
- Test defining and using tgrep2 macros.
525
- '''
526
- tree = ParentedTree.fromstring(
527
- '(VP (VB sold) (NP (DET the) '
528
- '(NN heiress)) (NP (NN deed) (PREP to) '
529
- '(NP (DET the) (NN school) (NN house))))')
530
- self.assertEqual(list(tgrep.tgrep_positions(
531
- '@ NP /^NP/;\n@ NN /^NN/;\n@NP !< @NP !$.. @NN',
532
- [tree])),
533
- [[(1,), (2, 2)]])
534
- # use undefined macro @CNP
535
- self.assertRaises(
536
- tgrep.TgrepException,
537
- list,
538
- tgrep.tgrep_positions(
539
- '@ NP /^NP/;\n@ NN /^NN/;\n@CNP !< @NP !$.. @NN', [tree]))
540
-
541
- def test_tokenize_node_labels(self):
542
- '''Test tokenization of labeled nodes.'''
543
- self.assertEqual(tgrep.tgrep_tokenize(
544
- 'S < @SBJ < (@VP < (@VB $.. @OBJ))'),
545
- ['S', '<', '@SBJ', '<', '(', '@VP', '<', '(',
546
- '@VB', '$..', '@OBJ', ')', ')'])
547
- self.assertEqual(tgrep.tgrep_tokenize(
548
- 'S < @SBJ=s < (@VP=v < (@VB $.. @OBJ))'),
549
- ['S', '<', '@SBJ', '=', 's', '<', '(', '@VP',
550
- '=', 'v', '<', '(', '@VB', '$..', '@OBJ', ')',
551
- ')'])
552
-
553
- def test_tokenize_segmented_patterns(self):
554
- '''Test tokenization of segmented patterns.'''
555
- self.assertEqual(tgrep.tgrep_tokenize(
556
- 'S < @SBJ=s < (@VP=v < (@VB $.. @OBJ)) : =s .. =v'),
557
- ['S', '<', '@SBJ', '=', 's', '<', '(', '@VP',
558
- '=', 'v', '<', '(', '@VB', '$..', '@OBJ', ')',
559
- ')', ':', '=s', '..', '=v'])
560
-
561
- def test_labeled_nodes(self):
562
- '''
563
- Test labeled nodes.
564
-
565
- Test case from Emily M. Bender.
566
- '''
567
- search = '''
568
- # macros
569
- @ SBJ /SBJ/;
570
- @ VP /VP/;
571
- @ VB /VB/;
572
- @ VPoB /V[PB]/;
573
- @ OBJ /OBJ/;
574
-
575
- # 1 svo
576
- S < @SBJ=s < (@VP=v < (@VB $.. @OBJ)) : =s .. =v'''
577
- sent1 = ParentedTree.fromstring(
578
- '(S (NP-SBJ I) (VP (VB eat) (NP-OBJ (NNS apples))))')
579
- sent2 = ParentedTree.fromstring(
580
- '(S (VP (VB eat) (NP-OBJ (NNS apples))) (NP-SBJ I))')
581
- search_firsthalf = (search.split('\n\n')[0] +
582
- 'S < @SBJ < (@VP < (@VB $.. @OBJ))')
583
- search_rewrite = 'S < (/.*SBJ/ $.. (/VP/ < (/VB/ $.. /.*OBJ/)))'
584
-
585
- self.assertTrue(list(tgrep.tgrep_positions(search_firsthalf, [sent1]))[0])
586
- self.assertTrue(list(tgrep.tgrep_positions(search, [sent1]))[0])
587
- self.assertTrue(list(tgrep.tgrep_positions(search_rewrite, [sent1]))[0])
588
- self.assertEqual(list(tgrep.tgrep_positions(search, [sent1])),
589
- list(tgrep.tgrep_positions(search_rewrite, [sent1])))
590
- self.assertTrue(list(tgrep.tgrep_positions(search_firsthalf, [sent2]))[0])
591
- self.assertFalse(list(tgrep.tgrep_positions(search, [sent2]))[0])
592
- self.assertFalse(list(tgrep.tgrep_positions(search_rewrite, [sent2]))[0])
593
- self.assertEqual(list(tgrep.tgrep_positions(search, [sent2])),
594
- list(tgrep.tgrep_positions(search_rewrite, [sent2])))
595
-
596
- def test_multiple_conjs(self):
597
- '''
598
- Test that multiple (3 or more) conjunctions of node relations are
599
- handled properly.
600
- '''
601
- sent = ParentedTree.fromstring(
602
- '((A (B b) (C c)) (A (B b) (C c) (D d)))')
603
- # search = '(A < B < C < D)'
604
- # search_tworels = '(A < B < C)'
605
- self.assertEqual(list(tgrep.tgrep_positions('(A < B < C < D)', [sent])),
606
- [[(1,)]])
607
- self.assertEqual(list(tgrep.tgrep_positions('(A < B < C)', [sent])),
608
- [[(0,), (1,)]])
609
-
610
- def test_trailing_semicolon(self):
611
- '''
612
- Test that semicolons at the end of a tgrep2 search string won't
613
- cause a parse failure.
614
- '''
615
- tree = ParentedTree.fromstring(
616
- '(S (NP (DT the) (JJ big) (NN dog)) '
617
- '(VP bit) (NP (DT a) (NN cat)))')
618
- self.assertEqual(list(tgrep.tgrep_positions('NN', [tree])),
619
- [[(0,2), (2,1)]])
620
- self.assertEqual(list(tgrep.tgrep_positions('NN;', [tree])),
621
- [[(0,2), (2,1)]])
622
- self.assertEqual(list(tgrep.tgrep_positions('NN;;', [tree])),
623
- [[(0,2), (2,1)]])
624
-
625
- if __name__ == '__main__':
626
- unittest.main()
1
+ #!/usr/bin/env python
2
+ # -*- coding: utf-8 -*-
3
+ #
4
+ # Natural Language Toolkit: TGrep search
5
+ #
6
+ # Copyright (C) 2001-2018 NLTK Project
7
+ # Author: Will Roberts <wildwilhelm@gmail.com>
8
+ # URL: <http://nltk.org/>
9
+ # For license information, see LICENSE.TXT
10
+
11
+ '''
12
+ Unit tests for nltk.tgrep.
13
+ '''
14
+
15
+ from __future__ import absolute_import, print_function, unicode_literals
16
+
17
+ from six import b
18
+
19
+ from nltk.tree import ParentedTree
20
+ from nltk import tgrep
21
+ import unittest
22
+
23
+ class TestSequenceFunctions(unittest.TestCase):
24
+
25
+ '''
26
+ Class containing unit tests for nltk.tgrep.
27
+ '''
28
+
29
+ def test_tokenize_simple(self):
30
+ '''
31
+ Simple test of tokenization.
32
+ '''
33
+ tokens = tgrep.tgrep_tokenize('A .. (B !< C . D) | ![<< (E , F) $ G]')
34
+ self.assertEqual(tokens,
35
+ ['A', '..', '(', 'B', '!', '<', 'C', '.', 'D', ')',
36
+ '|', '!', '[', '<<', '(', 'E', ',', 'F', ')', '$',
37
+ 'G', ']'])
38
+
39
+ def test_tokenize_encoding(self):
40
+ '''
41
+ Test that tokenization handles bytes and strs the same way.
42
+ '''
43
+ self.assertEqual(
44
+ tgrep.tgrep_tokenize(b('A .. (B !< C . D) | ![<< (E , F) $ G]')),
45
+ tgrep.tgrep_tokenize('A .. (B !< C . D) | ![<< (E , F) $ G]'))
46
+
47
+ def test_tokenize_link_types(self):
48
+ '''
49
+ Test tokenization of basic link types.
50
+ '''
51
+ self.assertEqual(tgrep.tgrep_tokenize('A<B'), ['A', '<', 'B'])
52
+ self.assertEqual(tgrep.tgrep_tokenize('A>B'), ['A', '>', 'B'])
53
+ self.assertEqual(tgrep.tgrep_tokenize('A<3B'), ['A', '<3', 'B'])
54
+ self.assertEqual(tgrep.tgrep_tokenize('A>3B'), ['A', '>3', 'B'])
55
+ self.assertEqual(tgrep.tgrep_tokenize('A<,B'), ['A', '<,', 'B'])
56
+ self.assertEqual(tgrep.tgrep_tokenize('A>,B'), ['A', '>,', 'B'])
57
+ self.assertEqual(tgrep.tgrep_tokenize('A<-3B'), ['A', '<-3', 'B'])
58
+ self.assertEqual(tgrep.tgrep_tokenize('A>-3B'), ['A', '>-3', 'B'])
59
+ self.assertEqual(tgrep.tgrep_tokenize('A<-B'), ['A', '<-', 'B'])
60
+ self.assertEqual(tgrep.tgrep_tokenize('A>-B'), ['A', '>-', 'B'])
61
+ self.assertEqual(tgrep.tgrep_tokenize('A<\'B'), ['A', '<\'', 'B'])
62
+ self.assertEqual(tgrep.tgrep_tokenize('A>\'B'), ['A', '>\'', 'B'])
63
+ self.assertEqual(tgrep.tgrep_tokenize('A<:B'), ['A', '<:', 'B'])
64
+ self.assertEqual(tgrep.tgrep_tokenize('A>:B'), ['A', '>:', 'B'])
65
+ self.assertEqual(tgrep.tgrep_tokenize('A<<B'), ['A', '<<', 'B'])
66
+ self.assertEqual(tgrep.tgrep_tokenize('A>>B'), ['A', '>>', 'B'])
67
+ self.assertEqual(tgrep.tgrep_tokenize('A<<,B'), ['A', '<<,', 'B'])
68
+ self.assertEqual(tgrep.tgrep_tokenize('A>>,B'), ['A', '>>,', 'B'])
69
+ self.assertEqual(tgrep.tgrep_tokenize('A<<\'B'), ['A', '<<\'', 'B'])
70
+ self.assertEqual(tgrep.tgrep_tokenize('A>>\'B'), ['A', '>>\'', 'B'])
71
+ self.assertEqual(tgrep.tgrep_tokenize('A<<:B'), ['A', '<<:', 'B'])
72
+ self.assertEqual(tgrep.tgrep_tokenize('A>>:B'), ['A', '>>:', 'B'])
73
+ self.assertEqual(tgrep.tgrep_tokenize('A.B'), ['A', '.', 'B'])
74
+ self.assertEqual(tgrep.tgrep_tokenize('A,B'), ['A', ',', 'B'])
75
+ self.assertEqual(tgrep.tgrep_tokenize('A..B'), ['A', '..', 'B'])
76
+ self.assertEqual(tgrep.tgrep_tokenize('A,,B'), ['A', ',,', 'B'])
77
+ self.assertEqual(tgrep.tgrep_tokenize('A$B'), ['A', '$', 'B'])
78
+ self.assertEqual(tgrep.tgrep_tokenize('A$.B'), ['A', '$.', 'B'])
79
+ self.assertEqual(tgrep.tgrep_tokenize('A$,B'), ['A', '$,', 'B'])
80
+ self.assertEqual(tgrep.tgrep_tokenize('A$..B'), ['A', '$..', 'B'])
81
+ self.assertEqual(tgrep.tgrep_tokenize('A$,,B'), ['A', '$,,', 'B'])
82
+ self.assertEqual(tgrep.tgrep_tokenize('A!<B'), ['A', '!', '<', 'B'])
83
+ self.assertEqual(tgrep.tgrep_tokenize('A!>B'), ['A', '!', '>', 'B'])
84
+ self.assertEqual(tgrep.tgrep_tokenize('A!<3B'), ['A', '!', '<3', 'B'])
85
+ self.assertEqual(tgrep.tgrep_tokenize('A!>3B'), ['A', '!', '>3', 'B'])
86
+ self.assertEqual(tgrep.tgrep_tokenize('A!<,B'), ['A', '!', '<,', 'B'])
87
+ self.assertEqual(tgrep.tgrep_tokenize('A!>,B'), ['A', '!', '>,', 'B'])
88
+ self.assertEqual(tgrep.tgrep_tokenize('A!<-3B'),
89
+ ['A', '!', '<-3', 'B'])
90
+ self.assertEqual(tgrep.tgrep_tokenize('A!>-3B'),
91
+ ['A', '!', '>-3', 'B'])
92
+ self.assertEqual(tgrep.tgrep_tokenize('A!<-B'), ['A', '!', '<-', 'B'])
93
+ self.assertEqual(tgrep.tgrep_tokenize('A!>-B'), ['A', '!', '>-', 'B'])
94
+ self.assertEqual(tgrep.tgrep_tokenize('A!<\'B'),
95
+ ['A', '!', '<\'', 'B'])
96
+ self.assertEqual(tgrep.tgrep_tokenize('A!>\'B'),
97
+ ['A', '!', '>\'', 'B'])
98
+ self.assertEqual(tgrep.tgrep_tokenize('A!<:B'), ['A', '!', '<:', 'B'])
99
+ self.assertEqual(tgrep.tgrep_tokenize('A!>:B'), ['A', '!', '>:', 'B'])
100
+ self.assertEqual(tgrep.tgrep_tokenize('A!<<B'), ['A', '!', '<<', 'B'])
101
+ self.assertEqual(tgrep.tgrep_tokenize('A!>>B'), ['A', '!', '>>', 'B'])
102
+ self.assertEqual(tgrep.tgrep_tokenize('A!<<,B'),
103
+ ['A', '!', '<<,', 'B'])
104
+ self.assertEqual(tgrep.tgrep_tokenize('A!>>,B'),
105
+ ['A', '!', '>>,', 'B'])
106
+ self.assertEqual(tgrep.tgrep_tokenize('A!<<\'B'),
107
+ ['A', '!', '<<\'', 'B'])
108
+ self.assertEqual(tgrep.tgrep_tokenize('A!>>\'B'),
109
+ ['A', '!', '>>\'', 'B'])
110
+ self.assertEqual(tgrep.tgrep_tokenize('A!<<:B'),
111
+ ['A', '!', '<<:', 'B'])
112
+ self.assertEqual(tgrep.tgrep_tokenize('A!>>:B'),
113
+ ['A', '!', '>>:', 'B'])
114
+ self.assertEqual(tgrep.tgrep_tokenize('A!.B'), ['A', '!', '.', 'B'])
115
+ self.assertEqual(tgrep.tgrep_tokenize('A!,B'), ['A', '!', ',', 'B'])
116
+ self.assertEqual(tgrep.tgrep_tokenize('A!..B'), ['A', '!', '..', 'B'])
117
+ self.assertEqual(tgrep.tgrep_tokenize('A!,,B'), ['A', '!', ',,', 'B'])
118
+ self.assertEqual(tgrep.tgrep_tokenize('A!$B'), ['A', '!', '$', 'B'])
119
+ self.assertEqual(tgrep.tgrep_tokenize('A!$.B'), ['A', '!', '$.', 'B'])
120
+ self.assertEqual(tgrep.tgrep_tokenize('A!$,B'), ['A', '!', '$,', 'B'])
121
+ self.assertEqual(tgrep.tgrep_tokenize('A!$..B'),
122
+ ['A', '!', '$..', 'B'])
123
+ self.assertEqual(tgrep.tgrep_tokenize('A!$,,B'),
124
+ ['A', '!', '$,,', 'B'])
125
+
126
+ def test_tokenize_examples(self):
127
+ '''
128
+ Test tokenization of the TGrep2 manual example patterns.
129
+ '''
130
+ self.assertEqual(tgrep.tgrep_tokenize('NP < PP'),
131
+ ['NP', '<', 'PP'])
132
+ self.assertEqual(tgrep.tgrep_tokenize('/^NP/'),
133
+ ['/^NP/'])
134
+ self.assertEqual(tgrep.tgrep_tokenize('NP << PP . VP'),
135
+ ['NP', '<<', 'PP', '.', 'VP'])
136
+ self.assertEqual(tgrep.tgrep_tokenize('NP << PP | . VP'),
137
+ ['NP', '<<', 'PP', '|', '.', 'VP'])
138
+ self.assertEqual(tgrep.tgrep_tokenize('NP !<< PP [> NP | >> VP]'),
139
+ ['NP', '!', '<<', 'PP', '[', '>', 'NP', '|',
140
+ '>>', 'VP', ']'])
141
+ self.assertEqual(tgrep.tgrep_tokenize('NP << (PP . VP)'),
142
+ ['NP', '<<', '(', 'PP', '.', 'VP', ')'])
143
+ self.assertEqual(tgrep.tgrep_tokenize('NP <\' (PP <, (IN < on))'),
144
+ ['NP', '<\'', '(', 'PP', '<,', '(', 'IN', '<',
145
+ 'on', ')', ')'])
146
+ self.assertEqual(tgrep.tgrep_tokenize('S < (A < B) < C'),
147
+ ['S', '<', '(', 'A', '<', 'B', ')', '<', 'C'])
148
+ self.assertEqual(tgrep.tgrep_tokenize('S < ((A < B) < C)'),
149
+ ['S', '<', '(', '(', 'A', '<', 'B', ')',
150
+ '<', 'C', ')'])
151
+ self.assertEqual(tgrep.tgrep_tokenize('S < (A < B < C)'),
152
+ ['S', '<', '(', 'A', '<', 'B', '<', 'C', ')'])
153
+ self.assertEqual(tgrep.tgrep_tokenize('A<B&.C'),
154
+ ['A', '<', 'B', '&', '.', 'C'])
155
+
156
+ def test_tokenize_quoting(self):
157
+ '''
158
+ Test tokenization of quoting.
159
+ '''
160
+ self.assertEqual(tgrep.tgrep_tokenize('"A<<:B"<<:"A $.. B"<"A>3B"<C'),
161
+ ['"A<<:B"', '<<:', '"A $.. B"', '<', '"A>3B"',
162
+ '<', 'C'])
163
+
164
+ def test_tokenize_nodenames(self):
165
+ '''
166
+ Test tokenization of node names.
167
+ '''
168
+ self.assertEqual(tgrep.tgrep_tokenize('Robert'), ['Robert'])
169
+ self.assertEqual(tgrep.tgrep_tokenize('/^[Bb]ob/'), ['/^[Bb]ob/'])
170
+ self.assertEqual(tgrep.tgrep_tokenize('*'), ['*'])
171
+ self.assertEqual(tgrep.tgrep_tokenize('__'), ['__'])
172
+ # test tokenization of NLTK tree position syntax
173
+ self.assertEqual(tgrep.tgrep_tokenize('N()'),
174
+ ['N(', ')'])
175
+ self.assertEqual(tgrep.tgrep_tokenize('N(0,)'),
176
+ ['N(', '0', ',', ')'])
177
+ self.assertEqual(tgrep.tgrep_tokenize('N(0,0)'),
178
+ ['N(', '0', ',', '0', ')'])
179
+ self.assertEqual(tgrep.tgrep_tokenize('N(0,0,)'),
180
+ ['N(', '0', ',', '0', ',', ')'])
181
+
182
+ def test_tokenize_macros(self):
183
+ '''
184
+ Test tokenization of macro definitions.
185
+ '''
186
+ self.assertEqual(tgrep.tgrep_tokenize(
187
+ '@ NP /^NP/;\n@ NN /^NN/;\n@NP [!< NP | < @NN] !$.. @NN'),
188
+ ['@', 'NP', '/^NP/', ';', '@', 'NN', '/^NN/', ';',
189
+ '@NP', '[', '!', '<', 'NP', '|', '<', '@NN', ']',
190
+ '!', '$..', '@NN'])
191
+
192
+ def test_node_simple(self):
193
+ '''
194
+ Test a simple use of tgrep for finding nodes matching a given
195
+ pattern.
196
+ '''
197
+ tree = ParentedTree.fromstring(
198
+ '(S (NP (DT the) (JJ big) (NN dog)) '
199
+ '(VP bit) (NP (DT a) (NN cat)))')
200
+ self.assertEqual(list(tgrep.tgrep_positions('NN', [tree])),
201
+ [[(0,2), (2,1)]])
202
+ self.assertEqual(list(tgrep.tgrep_nodes('NN', [tree])),
203
+ [[tree[0,2], tree[2,1]]])
204
+ self.assertEqual(list(tgrep.tgrep_positions('NN|JJ', [tree])),
205
+ [[(0, 1), (0, 2), (2, 1)]])
206
+
207
+ def test_node_printing(self):
208
+ '''Test that the tgrep print operator ' is properly ignored.'''
209
+ tree = ParentedTree.fromstring('(S (n x) (N x))')
210
+ self.assertEqual(list(tgrep.tgrep_positions('N', [tree])),
211
+ list(tgrep.tgrep_positions('\'N', [tree])))
212
+ self.assertEqual(list(tgrep.tgrep_positions('/[Nn]/', [tree])),
213
+ list(tgrep.tgrep_positions('\'/[Nn]/', [tree])))
214
+
215
+ def test_node_encoding(self):
216
+ '''
217
+ Test that tgrep search strings handles bytes and strs the same
218
+ way.
219
+ '''
220
+ tree = ParentedTree.fromstring(
221
+ '(S (NP (DT the) (JJ big) (NN dog)) '
222
+ '(VP bit) (NP (DT a) (NN cat)))')
223
+ self.assertEqual(list(tgrep.tgrep_positions(b('NN'), [tree])),
224
+ list(tgrep.tgrep_positions('NN', [tree])))
225
+ self.assertEqual(list(tgrep.tgrep_nodes(b('NN'), [tree])),
226
+ list(tgrep.tgrep_nodes('NN', [tree])))
227
+ self.assertEqual(list(tgrep.tgrep_positions(b('NN|JJ'), [tree])),
228
+ list(tgrep.tgrep_positions('NN|JJ', [tree])))
229
+
230
+ def test_node_nocase(self):
231
+ '''
232
+ Test selecting nodes using case insensitive node names.
233
+ '''
234
+ tree = ParentedTree.fromstring('(S (n x) (N x))')
235
+ self.assertEqual(list(tgrep.tgrep_positions('"N"', [tree])), [[(1,)]])
236
+ self.assertEqual(list(tgrep.tgrep_positions('i@"N"', [tree])), [[(0,), (1,)]])
237
+
238
+ def test_node_quoted(self):
239
+ '''
240
+ Test selecting nodes using quoted node names.
241
+ '''
242
+ tree = ParentedTree.fromstring('(N ("N" x) (N" x) ("\\" x))')
243
+ self.assertEqual(list(tgrep.tgrep_positions('"N"', [tree])), [[()]])
244
+ self.assertEqual(list(tgrep.tgrep_positions('"\\"N\\""', [tree])), [[(0,)]])
245
+ self.assertEqual(list(tgrep.tgrep_positions('"N\\""', [tree])), [[(1,)]])
246
+ self.assertEqual(list(tgrep.tgrep_positions('"\\"\\\\\\""', [tree])), [[(2,)]])
247
+
248
+ def test_node_regex(self):
249
+ '''
250
+ Test regex matching on nodes.
251
+ '''
252
+ tree = ParentedTree.fromstring('(S (NP-SBJ x) (NP x) (NNP x) (VP x))')
253
+ # This is a regular expression that matches any node whose
254
+ # name starts with NP, including NP-SBJ:
255
+ self.assertEqual(list(tgrep.tgrep_positions('/^NP/', [tree])),
256
+ [[(0,), (1,)]])
257
+
258
+ def test_node_regex_2(self):
259
+ '''
260
+ Test regex matching on nodes.
261
+ '''
262
+ tree = ParentedTree.fromstring('(S (SBJ x) (SBJ1 x) (NP-SBJ x))')
263
+ self.assertEqual(list(tgrep.tgrep_positions('/^SBJ/', [tree])),
264
+ [[(0,), (1,)]])
265
+ # This is a regular expression that matches any node whose
266
+ # name includes SBJ, including NP-SBJ:
267
+ self.assertEqual(list(tgrep.tgrep_positions('/SBJ/', [tree])),
268
+ [[(0,), (1,), (2,)]])
269
+
270
+ def test_node_tree_position(self):
271
+ '''
272
+ Test matching on nodes based on NLTK tree position.
273
+ '''
274
+ tree = ParentedTree.fromstring('(S (NP-SBJ x) (NP x) (NNP x) (VP x))')
275
+ # test all tree positions that are not leaves
276
+ leaf_positions = set([tree.leaf_treeposition(x)
277
+ for x in range(len(tree.leaves()))])
278
+ tree_positions = [x for x in tree.treepositions()
279
+ if x not in leaf_positions]
280
+ for position in tree_positions:
281
+ node_id = 'N{0}'.format(position)
282
+ tgrep_positions = list(tgrep.tgrep_positions(node_id, [tree]))
283
+ self.assertEqual(len(tgrep_positions[0]), 1)
284
+ self.assertEqual(tgrep_positions[0][0], position)
285
+
286
+ def test_node_noleaves(self):
287
+ '''
288
+ Test node name matching with the search_leaves flag set to False.
289
+ '''
290
+ tree = ParentedTree.fromstring('(S (A (T x)) (B (N x)))')
291
+ self.assertEqual(list(tgrep.tgrep_positions('x', [tree])),
292
+ [[(0, 0, 0), (1, 0, 0)]])
293
+ self.assertEqual(list(tgrep.tgrep_positions('x', [tree], False)),
294
+ [[]])
295
+
296
+ def tests_rel_dominance(self):
297
+ '''
298
+ Test matching nodes based on dominance relations.
299
+ '''
300
+ tree = ParentedTree.fromstring('(S (A (T x)) (B (N x)))')
301
+ self.assertEqual(list(tgrep.tgrep_positions('* < T', [tree])),
302
+ [[(0,)]])
303
+ self.assertEqual(list(tgrep.tgrep_positions('* < T > S', [tree])),
304
+ [[(0,)]])
305
+ self.assertEqual(list(tgrep.tgrep_positions('* !< T', [tree])),
306
+ [[(), (0, 0), (0, 0, 0), (1,), (1, 0), (1, 0, 0)]])
307
+ self.assertEqual(list(tgrep.tgrep_positions('* !< T > S', [tree])),
308
+ [[(1,)]])
309
+ self.assertEqual(list(tgrep.tgrep_positions('* > A', [tree])),
310
+ [[(0, 0)]])
311
+ self.assertEqual(list(tgrep.tgrep_positions('* > B', [tree])),
312
+ [[(1, 0)]])
313
+ self.assertEqual(list(tgrep.tgrep_positions('* !> B', [tree])),
314
+ [[(), (0,), (0, 0), (0, 0, 0), (1,), (1, 0, 0)]])
315
+ self.assertEqual(list(tgrep.tgrep_positions('* !> B >> S', [tree])),
316
+ [[(0,), (0, 0), (1,)]])
317
+ self.assertEqual(list(tgrep.tgrep_positions('* >> S', [tree])),
318
+ [[(0,), (0, 0), (1,), (1, 0)]])
319
+ self.assertEqual(list(tgrep.tgrep_positions('* >>, S', [tree])),
320
+ [[(0,), (0, 0)]])
321
+ self.assertEqual(list(tgrep.tgrep_positions('* >>\' S', [tree])),
322
+ [[(1,), (1, 0)]])
323
+ # Known issue:
324
+ #self.assertEqual(list(tgrep.tgrep_positions('* !>> S', [tree])),
325
+ # [[()]])
326
+ self.assertEqual(list(tgrep.tgrep_positions('* << T', [tree])),
327
+ [[(), (0,)]])
328
+ self.assertEqual(list(tgrep.tgrep_positions('* <<\' T', [tree])),
329
+ [[(0,)]])
330
+ self.assertEqual(list(tgrep.tgrep_positions('* <<1 N', [tree])),
331
+ [[(1,)]])
332
+ self.assertEqual(list(tgrep.tgrep_positions('* !<< T', [tree])),
333
+ [[(0, 0), (0, 0, 0), (1,), (1, 0), (1, 0, 0)]])
334
+ tree = ParentedTree.fromstring('(S (A (T x)) (B (T x) (N x )))')
335
+ self.assertEqual(list(tgrep.tgrep_positions('* <: T', [tree])),
336
+ [[(0,)]])
337
+ self.assertEqual(list(tgrep.tgrep_positions('* < T', [tree])),
338
+ [[(0,), (1,)]])
339
+ self.assertEqual(list(tgrep.tgrep_positions('* !<: T', [tree])),
340
+ [[(), (0, 0), (0, 0, 0), (1,), (1, 0), (1, 0, 0),
341
+ (1, 1), (1, 1, 0)]])
342
+ self.assertEqual(list(tgrep.tgrep_positions('* !<: T > S', [tree])),
343
+ [[(1,)]])
344
+ tree = ParentedTree.fromstring('(S (T (A x) (B x)) (T (C x)))')
345
+ self.assertEqual(list(tgrep.tgrep_positions('* >: T', [tree])),
346
+ [[(1, 0)]])
347
+ self.assertEqual(list(tgrep.tgrep_positions('* !>: T', [tree])),
348
+ [[(), (0,), (0, 0), (0, 0, 0), (0, 1), (0, 1, 0),
349
+ (1,), (1, 0, 0)]])
350
+ tree = ParentedTree.fromstring('(S (A (B (C (D (E (T x))))))'
351
+ ' (A (B (C (D (E (T x))) (N x)))))')
352
+ self.assertEqual(list(tgrep.tgrep_positions('* <<: T', [tree])),
353
+ [[(0,), (0, 0), (0, 0, 0), (0, 0, 0, 0),
354
+ (0, 0, 0, 0, 0), (1, 0, 0, 0), (1, 0, 0, 0, 0)]])
355
+ self.assertEqual(list(tgrep.tgrep_positions('* >>: A', [tree])),
356
+ [[(0, 0), (0, 0, 0), (0, 0, 0, 0), (0, 0, 0, 0, 0),
357
+ (0, 0, 0, 0, 0, 0), (1, 0), (1, 0, 0)]])
358
+
359
+ def test_bad_operator(self):
360
+ '''
361
+ Test error handling of undefined tgrep operators.
362
+ '''
363
+ tree = ParentedTree.fromstring('(S (A (T x)) (B (N x)))')
364
+ self.assertRaises(
365
+ tgrep.TgrepException,
366
+ list,
367
+ tgrep.tgrep_positions('* >>> S', [tree]))
368
+
369
+ def test_comments(self):
370
+ '''
371
+ Test that comments are correctly filtered out of tgrep search
372
+ strings.
373
+ '''
374
+ tree = ParentedTree.fromstring('(S (NN x) (NP x) (NN x))')
375
+ search1 = '''
376
+ @ NP /^NP/;
377
+ @ NN /^NN/;
378
+ @NN
379
+ '''
380
+ self.assertEqual(list(tgrep.tgrep_positions(search1, [tree])),
381
+ [[(0,), (2,)]])
382
+ search2 = '''
383
+ # macros
384
+ @ NP /^NP/;
385
+ @ NN /^NN/;
386
+
387
+ # search string
388
+ @NN
389
+ '''
390
+ self.assertEqual(list(tgrep.tgrep_positions(search2, [tree])),
391
+ [[(0,), (2,)]])
392
+
393
+ def test_rel_sister_nodes(self):
394
+ '''
395
+ Test matching sister nodes in a tree.
396
+ '''
397
+ tree = ParentedTree.fromstring('(S (A x) (B x) (C x))')
398
+ self.assertEqual(list(tgrep.tgrep_positions('* $. B', [tree])), [[(0,)]])
399
+ self.assertEqual(list(tgrep.tgrep_positions('* $.. B', [tree])), [[(0,)]])
400
+ self.assertEqual(list(tgrep.tgrep_positions('* $, B', [tree])), [[(2,)]])
401
+ self.assertEqual(list(tgrep.tgrep_positions('* $,, B', [tree])), [[(2,)]])
402
+ self.assertEqual(list(tgrep.tgrep_positions('* $ B', [tree])), [[(0,), (2,)]])
403
+
404
+ def tests_rel_indexed_children(self):
405
+ '''
406
+ Test matching nodes based on their index in their parent node.
407
+ '''
408
+ tree = ParentedTree.fromstring('(S (A x) (B x) (C x))')
409
+ self.assertEqual(list(tgrep.tgrep_positions('* >, S', [tree])), [[(0,)]])
410
+ self.assertEqual(list(tgrep.tgrep_positions('* >1 S', [tree])), [[(0,)]])
411
+ self.assertEqual(list(tgrep.tgrep_positions('* >2 S', [tree])), [[(1,)]])
412
+ self.assertEqual(list(tgrep.tgrep_positions('* >3 S', [tree])), [[(2,)]])
413
+ self.assertEqual(list(tgrep.tgrep_positions('* >\' S', [tree])), [[(2,)]])
414
+ self.assertEqual(list(tgrep.tgrep_positions('* >-1 S', [tree])), [[(2,)]])
415
+ self.assertEqual(list(tgrep.tgrep_positions('* >-2 S', [tree])), [[(1,)]])
416
+ self.assertEqual(list(tgrep.tgrep_positions('* >-3 S', [tree])), [[(0,)]])
417
+ tree = ParentedTree.fromstring(
418
+ '(S (D (A x) (B x) (C x)) (E (B x) (C x) (A x)) '
419
+ '(F (C x) (A x) (B x)))')
420
+ self.assertEqual(list(tgrep.tgrep_positions('* <, A', [tree])), [[(0,)]])
421
+ self.assertEqual(list(tgrep.tgrep_positions('* <1 A', [tree])), [[(0,)]])
422
+ self.assertEqual(list(tgrep.tgrep_positions('* <2 A', [tree])), [[(2,)]])
423
+ self.assertEqual(list(tgrep.tgrep_positions('* <3 A', [tree])), [[(1,)]])
424
+ self.assertEqual(list(tgrep.tgrep_positions('* <\' A', [tree])), [[(1,)]])
425
+ self.assertEqual(list(tgrep.tgrep_positions('* <-1 A', [tree])), [[(1,)]])
426
+ self.assertEqual(list(tgrep.tgrep_positions('* <-2 A', [tree])), [[(2,)]])
427
+ self.assertEqual(list(tgrep.tgrep_positions('* <-3 A', [tree])), [[(0,)]])
428
+
429
+ def test_rel_precedence(self):
430
+ '''
431
+ Test matching nodes based on precedence relations.
432
+ '''
433
+ tree = ParentedTree.fromstring('(S (NP (NP (PP x)) (NP (AP x)))'
434
+ ' (VP (AP (X (PP x)) (Y (AP x))))'
435
+ ' (NP (RC (NP (AP x)))))')
436
+ self.assertEqual(list(tgrep.tgrep_positions('* . X', [tree])),
437
+ [[(0,), (0, 1), (0, 1, 0)]])
438
+ self.assertEqual(list(tgrep.tgrep_positions('* . Y', [tree])),
439
+ [[(1, 0, 0), (1, 0, 0, 0)]])
440
+ self.assertEqual(list(tgrep.tgrep_positions('* .. X', [tree])),
441
+ [[(0,), (0, 0), (0, 0, 0), (0, 1), (0, 1, 0)]])
442
+ self.assertEqual(list(tgrep.tgrep_positions('* .. Y', [tree])),
443
+ [[(0,), (0, 0), (0, 0, 0), (0, 1), (0, 1, 0),
444
+ (1, 0, 0), (1, 0, 0, 0)]])
445
+ self.assertEqual(list(tgrep.tgrep_positions('* , X', [tree])),
446
+ [[(1, 0, 1), (1, 0, 1, 0)]])
447
+ self.assertEqual(list(tgrep.tgrep_positions('* , Y', [tree])),
448
+ [[(2,), (2, 0), (2, 0, 0), (2, 0, 0, 0)]])
449
+ self.assertEqual(list(tgrep.tgrep_positions('* ,, X', [tree])),
450
+ [[(1, 0, 1), (1, 0, 1, 0), (2,), (2, 0), (2, 0, 0),
451
+ (2, 0, 0, 0)]])
452
+ self.assertEqual(list(tgrep.tgrep_positions('* ,, Y', [tree])),
453
+ [[(2,), (2, 0), (2, 0, 0), (2, 0, 0, 0)]])
454
+
455
+ def test_examples(self):
456
+ '''
457
+ Test the Basic Examples from the TGrep2 manual.
458
+ '''
459
+ tree = ParentedTree.fromstring('(S (NP (AP x)) (NP (PP x)))')
460
+ # This matches any NP node that immediately dominates a PP:
461
+ self.assertEqual(list(tgrep.tgrep_positions('NP < PP', [tree])),
462
+ [[(1,)]])
463
+
464
+ tree = ParentedTree.fromstring('(S (NP x) (VP x) (NP (PP x)) (VP x))')
465
+ # This matches an NP that dominates a PP and is immediately
466
+ # followed by a VP:
467
+ self.assertEqual(list(tgrep.tgrep_positions('NP << PP . VP', [tree])),
468
+ [[(2,)]])
469
+
470
+ tree = ParentedTree.fromstring('(S (NP (AP x)) (NP (PP x)) '
471
+ '(NP (DET x) (NN x)) (VP x))')
472
+ # This matches an NP that dominates a PP or is immediately
473
+ # followed by a VP:
474
+ self.assertEqual(list(tgrep.tgrep_positions('NP << PP | . VP', [tree])),
475
+ [[(1,), (2,)]])
476
+
477
+ tree = ParentedTree.fromstring('(S (NP (NP (PP x)) (NP (AP x)))'
478
+ ' (VP (AP (NP (PP x)) (NP (AP x))))'
479
+ ' (NP (RC (NP (AP x)))))')
480
+ # This matches an NP that does not dominate a PP. Also, the NP
481
+ # must either have a parent that is an NP or be dominated by a
482
+ # VP:
483
+ self.assertEqual(list(tgrep.tgrep_positions(
484
+ 'NP !<< PP [> NP | >> VP]', [tree])),
485
+ [[(0, 1), (1, 0, 1)]])
486
+
487
+ tree = ParentedTree.fromstring('(S (NP (AP (PP x) (VP x))) '
488
+ '(NP (AP (PP x) (NP x))) (NP x))')
489
+ # This matches an NP that dominates a PP which itself is
490
+ # immediately followed by a VP. Note the use of parentheses to
491
+ # group ". VP" with the PP rather than with the NP:
492
+ self.assertEqual(list(tgrep.tgrep_positions('NP << (PP . VP)', [tree])),
493
+ [[(0,)]])
494
+
495
+ tree = ParentedTree.fromstring(
496
+ '(S (NP (DET a) (NN cat) (PP (IN on) (NP x)))'
497
+ ' (NP (DET a) (NN cat) (PP (IN on) (NP x)) (PP x))'
498
+ ' (NP x))')
499
+ # This matches an NP whose last child is a PP that begins with
500
+ # the preposition "on":
501
+ self.assertEqual(list(tgrep.tgrep_positions(
502
+ 'NP <\' (PP <, (IN < on))', [tree])),
503
+ [[(0,)]])
504
+
505
+ tree = ParentedTree.fromstring(
506
+ '(S (S (C x) (A (B x))) (S (C x) (A x)) '
507
+ '(S (D x) (A (B x))))')
508
+ # The following pattern matches an S which has a child A and
509
+ # another child that is a C and that the A has a child B:
510
+ self.assertEqual(list(tgrep.tgrep_positions('S < (A < B) < C', [tree])),
511
+ [[(0,)]])
512
+
513
+ tree = ParentedTree.fromstring(
514
+ '(S (S (A (B x) (C x))) (S (S (C x) (A (B x)))))')
515
+ # However, this pattern means that S has child A and that A
516
+ # has children B and C:
517
+ self.assertEqual(list(tgrep.tgrep_positions('S < ((A < B) < C)', [tree])),
518
+ [[(0,)]])
519
+
520
+ # It is equivalent to this:
521
+ self.assertEqual(list(tgrep.tgrep_positions('S < (A < B < C)', [tree])),
522
+ [[(0,)]])
523
+
524
+ def test_use_macros(self):
525
+ '''
526
+ Test defining and using tgrep2 macros.
527
+ '''
528
+ tree = ParentedTree.fromstring(
529
+ '(VP (VB sold) (NP (DET the) '
530
+ '(NN heiress)) (NP (NN deed) (PREP to) '
531
+ '(NP (DET the) (NN school) (NN house))))')
532
+ self.assertEqual(list(tgrep.tgrep_positions(
533
+ '@ NP /^NP/;\n@ NN /^NN/;\n@NP !< @NP !$.. @NN',
534
+ [tree])),
535
+ [[(1,), (2, 2)]])
536
+ # use undefined macro @CNP
537
+ self.assertRaises(
538
+ tgrep.TgrepException,
539
+ list,
540
+ tgrep.tgrep_positions(
541
+ '@ NP /^NP/;\n@ NN /^NN/;\n@CNP !< @NP !$.. @NN', [tree]))
542
+
543
+ def test_tokenize_node_labels(self):
544
+ '''Test tokenization of labeled nodes.'''
545
+ self.assertEqual(tgrep.tgrep_tokenize(
546
+ 'S < @SBJ < (@VP < (@VB $.. @OBJ))'),
547
+ ['S', '<', '@SBJ', '<', '(', '@VP', '<', '(',
548
+ '@VB', '$..', '@OBJ', ')', ')'])
549
+ self.assertEqual(tgrep.tgrep_tokenize(
550
+ 'S < @SBJ=s < (@VP=v < (@VB $.. @OBJ))'),
551
+ ['S', '<', '@SBJ', '=', 's', '<', '(', '@VP',
552
+ '=', 'v', '<', '(', '@VB', '$..', '@OBJ', ')',
553
+ ')'])
554
+
555
+ def test_tokenize_segmented_patterns(self):
556
+ '''Test tokenization of segmented patterns.'''
557
+ self.assertEqual(tgrep.tgrep_tokenize(
558
+ 'S < @SBJ=s < (@VP=v < (@VB $.. @OBJ)) : =s .. =v'),
559
+ ['S', '<', '@SBJ', '=', 's', '<', '(', '@VP',
560
+ '=', 'v', '<', '(', '@VB', '$..', '@OBJ', ')',
561
+ ')', ':', '=s', '..', '=v'])
562
+
563
+ def test_labeled_nodes(self):
564
+ '''
565
+ Test labeled nodes.
566
+
567
+ Test case from Emily M. Bender.
568
+ '''
569
+ search = '''
570
+ # macros
571
+ @ SBJ /SBJ/;
572
+ @ VP /VP/;
573
+ @ VB /VB/;
574
+ @ VPoB /V[PB]/;
575
+ @ OBJ /OBJ/;
576
+
577
+ # 1 svo
578
+ S < @SBJ=s < (@VP=v < (@VB $.. @OBJ)) : =s .. =v'''
579
+ sent1 = ParentedTree.fromstring(
580
+ '(S (NP-SBJ I) (VP (VB eat) (NP-OBJ (NNS apples))))')
581
+ sent2 = ParentedTree.fromstring(
582
+ '(S (VP (VB eat) (NP-OBJ (NNS apples))) (NP-SBJ I))')
583
+ search_firsthalf = (search.split('\n\n')[0] +
584
+ 'S < @SBJ < (@VP < (@VB $.. @OBJ))')
585
+ search_rewrite = 'S < (/.*SBJ/ $.. (/VP/ < (/VB/ $.. /.*OBJ/)))'
586
+
587
+ self.assertTrue(list(tgrep.tgrep_positions(search_firsthalf, [sent1]))[0])
588
+ self.assertTrue(list(tgrep.tgrep_positions(search, [sent1]))[0])
589
+ self.assertTrue(list(tgrep.tgrep_positions(search_rewrite, [sent1]))[0])
590
+ self.assertEqual(list(tgrep.tgrep_positions(search, [sent1])),
591
+ list(tgrep.tgrep_positions(search_rewrite, [sent1])))
592
+ self.assertTrue(list(tgrep.tgrep_positions(search_firsthalf, [sent2]))[0])
593
+ self.assertFalse(list(tgrep.tgrep_positions(search, [sent2]))[0])
594
+ self.assertFalse(list(tgrep.tgrep_positions(search_rewrite, [sent2]))[0])
595
+ self.assertEqual(list(tgrep.tgrep_positions(search, [sent2])),
596
+ list(tgrep.tgrep_positions(search_rewrite, [sent2])))
597
+
598
+ def test_multiple_conjs(self):
599
+ '''
600
+ Test that multiple (3 or more) conjunctions of node relations are
601
+ handled properly.
602
+ '''
603
+ sent = ParentedTree.fromstring(
604
+ '((A (B b) (C c)) (A (B b) (C c) (D d)))')
605
+ # search = '(A < B < C < D)'
606
+ # search_tworels = '(A < B < C)'
607
+ self.assertEqual(list(tgrep.tgrep_positions('(A < B < C < D)', [sent])),
608
+ [[(1,)]])
609
+ self.assertEqual(list(tgrep.tgrep_positions('(A < B < C)', [sent])),
610
+ [[(0,), (1,)]])
611
+
612
+ def test_trailing_semicolon(self):
613
+ '''
614
+ Test that semicolons at the end of a tgrep2 search string won't
615
+ cause a parse failure.
616
+ '''
617
+ tree = ParentedTree.fromstring(
618
+ '(S (NP (DT the) (JJ big) (NN dog)) '
619
+ '(VP bit) (NP (DT a) (NN cat)))')
620
+ self.assertEqual(list(tgrep.tgrep_positions('NN', [tree])),
621
+ [[(0,2), (2,1)]])
622
+ self.assertEqual(list(tgrep.tgrep_positions('NN;', [tree])),
623
+ [[(0,2), (2,1)]])
624
+ self.assertEqual(list(tgrep.tgrep_positions('NN;;', [tree])),
625
+ [[(0,2), (2,1)]])
626
+
627
+ if __name__ == '__main__':
628
+ unittest.main()