@@ 194-253 (lines=60) @@ | ||
191 | self.assertEqual(sullivan.subtreeType, 'undef') |
|
192 | self.assertEqual(sullivan.dfsTag, 0) |
|
193 | ||
194 | def testHierarchyConnectors2(self): |
|
195 | tree=computeTree(data.give_chief()) |
|
196 | NamedEntityMerging(tree).merge() |
|
197 | PrepositionMerging(tree).merge() |
|
198 | simplify(tree) |
|
199 | tree.sort() |
|
200 | root=tree |
|
201 | # Root |
|
202 | self.assertEqual(root.wordList, [Word("ROOT", 0)]) |
|
203 | self.assertEqual(root.namedEntityTag, 'undef') |
|
204 | self.assertEqual(root.dependency, 'R0') |
|
205 | self.assertEqual(root.parent, None) |
|
206 | self.assertEqual(len(root.child), 1) |
|
207 | self.assertEqual(root.subtreeType, 'PERSON') |
|
208 | self.assertEqual(root.dfsTag, 0) |
|
209 | # and |
|
210 | andw=root.child[0] |
|
211 | self.assertEqual(andw.wordList, [Word("and", 1000, None)]) |
|
212 | self.assertEqual(andw.namedEntityTag, 'undef') |
|
213 | self.assertEqual(andw.dependency, 'R0') |
|
214 | self.assertEqual(andw.parent, root) |
|
215 | self.assertEqual(len(andw.child), 2) |
|
216 | self.assertEqual(andw.subtreeType, 'PERSON') |
|
217 | self.assertEqual(andw.dfsTag, 0) |
|
218 | # identity |
|
219 | identity1=andw.child[0] |
|
220 | self.assertEqual(identity1.wordList, [Word("is", 2, 'VBZ')]) |
|
221 | self.assertEqual(identity1.namedEntityTag, 'undef') |
|
222 | self.assertEqual(identity1.dependency, 'RconjT') |
|
223 | self.assertEqual(identity1.parent, andw) |
|
224 | self.assertEqual(len(identity1.child), 1) |
|
225 | self.assertEqual(identity1.subtreeType, 'PERSON') |
|
226 | self.assertEqual(identity1.dfsTag, 0) |
|
227 | # identity |
|
228 | identity2=andw.child[1] |
|
229 | self.assertEqual(identity2.wordList, [Word("is", 2, 'VBZ')]) |
|
230 | self.assertEqual(identity2.namedEntityTag, 'undef') |
|
231 | self.assertEqual(identity2.dependency, 'RconjB') |
|
232 | self.assertEqual(identity2.parent, andw) |
|
233 | self.assertEqual(len(identity2.child), 1) |
|
234 | self.assertEqual(identity2.subtreeType, 'PERSON') |
|
235 | self.assertEqual(identity2.dfsTag, 0) |
|
236 | # chief |
|
237 | chief=identity1.child[0] |
|
238 | self.assertEqual(chief.wordList, [Word("chief", 4, 'NN')]) |
|
239 | self.assertEqual(chief.namedEntityTag, 'undef') |
|
240 | self.assertEqual(chief.dependency, 'R2') |
|
241 | self.assertEqual(chief.parent, identity1) |
|
242 | self.assertEqual(len(chief.child), 0) |
|
243 | self.assertEqual(chief.subtreeType, 'undef') |
|
244 | self.assertEqual(chief.dfsTag, 0) |
|
245 | # prime minister |
|
246 | prime=identity2.child[0] |
|
247 | self.assertEqual(prime.wordList, [Word("prime", 6, 'JJ'), Word("minister", 7, 'NN')]) |
|
248 | self.assertEqual(prime.namedEntityTag, 'undef') |
|
249 | self.assertEqual(prime.dependency, 'R2') |
|
250 | self.assertEqual(prime.parent, identity2) |
|
251 | self.assertEqual(len(prime.child), 0) |
|
252 | self.assertEqual(prime.subtreeType, 'undef') |
|
253 | self.assertEqual(prime.dfsTag, 0) |
|
254 | ||
255 | def testYesNoQuestion(self): |
|
256 | tree=computeTree(data.give_born()) |
@@ 166-223 (lines=58) @@ | ||
163 | self.assertEqual(the.subtreeType, 'undef') |
|
164 | self.assertEqual(the.dfsTag, 0) |
|
165 | ||
166 | def testNamedEntity2(self): |
|
167 | tree=computeTree(data.give_obama_president_usa()) |
|
168 | NamedEntityMerging(tree).merge() |
|
169 | tree.sort() |
|
170 | root=tree |
|
171 | # Root |
|
172 | self.assertEqual(root.wordList, [Word("ROOT", 0)]) |
|
173 | self.assertEqual(root.namedEntityTag, 'undef') |
|
174 | self.assertEqual(root.dependency, 'undef') |
|
175 | self.assertEqual(root.parent, None) |
|
176 | self.assertEqual(len(root.child), 1) |
|
177 | self.assertEqual(root.subtreeType, 'undef') |
|
178 | self.assertEqual(root.dfsTag, 0) |
|
179 | # Is |
|
180 | is_=root.child[0] |
|
181 | self.assertEqual(is_.wordList, [Word("is", 2, 'VBZ')]) |
|
182 | self.assertEqual(is_.namedEntityTag, 'undef') |
|
183 | self.assertEqual(is_.dependency, 'ROOT') |
|
184 | self.assertEqual(is_.parent, tree) |
|
185 | self.assertEqual(len(is_.child), 2) |
|
186 | self.assertEqual(is_.subtreeType, 'undef') |
|
187 | self.assertEqual(is_.dfsTag, 0) |
|
188 | # Obama |
|
189 | obama=is_.child[0] |
|
190 | self.assertEqual(obama.wordList, [Word("Obama", 1, 'NNP')]) |
|
191 | self.assertEqual(obama.namedEntityTag, 'PERSON') |
|
192 | self.assertEqual(obama.dependency, 'nsubj') |
|
193 | self.assertEqual(obama.parent, is_) |
|
194 | self.assertEqual(len(obama.child), 0) |
|
195 | self.assertEqual(obama.subtreeType, 'undef') |
|
196 | self.assertEqual(obama.dfsTag, 0) |
|
197 | # president |
|
198 | president =is_.child[1] |
|
199 | self.assertEqual(president.wordList, [Word("president", 6, 'NN')]) |
|
200 | self.assertEqual(president.namedEntityTag, 'undef') |
|
201 | self.assertEqual(president.dependency, 'xcomp') |
|
202 | self.assertEqual(president.parent, is_) |
|
203 | self.assertEqual(len(president.child), 2) |
|
204 | self.assertEqual(president.subtreeType, 'undef') |
|
205 | self.assertEqual(president.dfsTag, 0) |
|
206 | # The |
|
207 | the=president.child[0] |
|
208 | self.assertEqual(the.wordList, [Word("the", 3, 'DT')]) |
|
209 | self.assertEqual(the.namedEntityTag, 'undef') |
|
210 | self.assertEqual(the.dependency, 'det') |
|
211 | self.assertEqual(the.parent, president) |
|
212 | self.assertEqual(len(the.child), 0) |
|
213 | self.assertEqual(the.subtreeType, 'undef') |
|
214 | self.assertEqual(the.dfsTag, 0) |
|
215 | # United States |
|
216 | united=president.child[1] |
|
217 | self.assertEqual(united.wordList, [Word("United", 4, 'NNP'), Word("States", 5, 'NNPS')]) |
|
218 | self.assertEqual(united.namedEntityTag, 'LOCATION') |
|
219 | self.assertEqual(united.dependency, 'compound') |
|
220 | self.assertEqual(united.parent, president) |
|
221 | self.assertEqual(len(united.child), 0) |
|
222 | self.assertEqual(united.subtreeType, 'undef') |
|
223 | self.assertEqual(united.dfsTag, 0) |
|
224 | ||
225 | def testStr2(self): |
|
226 | tree=computeTree(data.give_john_smith()) |