@@ 357-400 (lines=44) @@ | ||
354 | self.assertEqual(obama.subtreeType, 'undef') |
|
355 | self.assertEqual(obama.dfsTag, 0) |
|
356 | ||
357 | def testPassIdentity(self): |
|
358 | tree=computeTree(data.mickey()) |
|
359 | NamedEntityMerging(tree).merge() |
|
360 | PrepositionMerging(tree).merge() |
|
361 | NamedEntityMerging(tree).merge() |
|
362 | PrepositionMerging(tree).merge() |
|
363 | simplify(tree) |
|
364 | tree.sort() |
|
365 | root=tree |
|
366 | # Root |
|
367 | self.assertEqual(root.wordList, [Word("ROOT", 0)]) |
|
368 | self.assertEqual(root.namedEntityTag, 'undef') |
|
369 | self.assertEqual(root.dependency, 'R0') |
|
370 | self.assertEqual(root.parent, None) |
|
371 | self.assertEqual(len(root.child), 1) |
|
372 | self.assertEqual(root.subtreeType, 'DATE') |
|
373 | self.assertEqual(root.dfsTag, 0) |
|
374 | # identity |
|
375 | identity=root.child[0] |
|
376 | self.assertEqual(identity.wordList, [Word("is", 2, 'VBZ')]) |
|
377 | self.assertEqual(identity.namedEntityTag, 'undef') |
|
378 | self.assertEqual(identity.dependency, 'R0') |
|
379 | self.assertEqual(identity.parent, root) |
|
380 | self.assertEqual(len(identity.child), 1) |
|
381 | self.assertEqual(identity.subtreeType, 'DATE') |
|
382 | self.assertEqual(identity.dfsTag, 0) |
|
383 | # birthday |
|
384 | birthday=identity.child[0] |
|
385 | self.assertEqual(birthday.wordList, [Word("birthday", 4, 'NN')]) |
|
386 | self.assertEqual(birthday.namedEntityTag, 'undef') |
|
387 | self.assertEqual(birthday.dependency, 'R2') |
|
388 | self.assertEqual(birthday.parent, identity) |
|
389 | self.assertEqual(len(birthday.child), 1) |
|
390 | self.assertEqual(birthday.subtreeType, 'undef') |
|
391 | self.assertEqual(birthday.dfsTag, 0) |
|
392 | # mickey |
|
393 | mickey=birthday.child[0] |
|
394 | self.assertEqual(mickey.wordList, [Word("Mickey", 6, 'NNP'), Word("Mouse", 7, 'NNP')]) |
|
395 | self.assertEqual(mickey.namedEntityTag, 'PERSON') |
|
396 | self.assertEqual(mickey.dependency, 'R2') |
|
397 | self.assertEqual(mickey.parent, birthday) |
|
398 | self.assertEqual(len(mickey.child), 0) |
|
399 | self.assertEqual(mickey.subtreeType, 'undef') |
|
400 | self.assertEqual(root.dfsTag, 0) |
|
401 | ||
@@ 22-63 (lines=42) @@ | ||
19 | PrepositionMerging(tree).merge() |
|
20 | self.assertEqual(simplify(tree), 'how old') |
|
21 | ||
22 | def testHierarchySimplification(self): |
|
23 | tree=computeTree(data.give_president_of_USA()) |
|
24 | NamedEntityMerging(tree).merge() |
|
25 | PrepositionMerging(tree).merge() |
|
26 | simplify(tree) |
|
27 | tree.sort() |
|
28 | root=tree |
|
29 | # Root |
|
30 | self.assertEqual(root.wordList, [Word("ROOT", 0)]) |
|
31 | self.assertEqual(root.namedEntityTag, 'undef') |
|
32 | self.assertEqual(root.dependency, 'R0') |
|
33 | self.assertEqual(root.parent, None) |
|
34 | self.assertEqual(len(root.child), 1) |
|
35 | self.assertEqual(root.subtreeType, 'PERSON') |
|
36 | self.assertEqual(root.dfsTag, 0) |
|
37 | # Is |
|
38 | is_=root.child[0] |
|
39 | self.assertEqual(is_.wordList, [Word("is", 2, 'VBZ')]) |
|
40 | self.assertEqual(is_.namedEntityTag, 'undef') |
|
41 | self.assertEqual(is_.dependency, 'R0') |
|
42 | self.assertEqual(is_.parent, root) |
|
43 | self.assertEqual(len(is_.child), 1) |
|
44 | self.assertEqual(is_.subtreeType, 'PERSON') |
|
45 | self.assertEqual(is_.dfsTag, 0) |
|
46 | # President |
|
47 | president=is_.child[0] |
|
48 | self.assertEqual(president.wordList, [Word("president", 4, 'NN')]) |
|
49 | self.assertEqual(president.namedEntityTag, 'undef') |
|
50 | self.assertEqual(president.dependency, 'R2') |
|
51 | self.assertEqual(president.parent, is_) |
|
52 | self.assertEqual(len(president.child), 1) |
|
53 | self.assertEqual(president.subtreeType, 'undef') |
|
54 | self.assertEqual(president.dfsTag, 0) |
|
55 | # United States |
|
56 | us=president.child[0] |
|
57 | self.assertEqual(us.wordList, [Word("United", 7, 'NNP'), Word("States", 8, 'NNPS')]) |
|
58 | self.assertEqual(us.namedEntityTag, 'LOCATION') |
|
59 | self.assertEqual(us.dependency, 'R2') |
|
60 | self.assertEqual(us.parent, president) |
|
61 | self.assertEqual(len(us.child), 0) |
|
62 | self.assertEqual(us.subtreeType, 'undef') |
|
63 | self.assertEqual(us.dfsTag, 0) |
|
64 | ||
65 | def testIgnore(self): |
|
66 | tree=computeTree(data.give_how_old()) |