Code Duplication    Length = 58-60 lines in 2 locations

tests/test_dependencyAnalysis.py 1 location

@@ 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())

tests/test_initialMerge.py 1 location

@@ 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())