Python Corpus.generate_neighbor_list Examples

Example #1

File: corpus_test.py Project: jgerrish/nltk_ext

 def test_generate_neighbor_list(self):
     corpus = Corpus([self.d1, self.d2, self.d3, self.d4])
     l = corpus.generate_neighbor_list(self.d1)
     self.assertTrue(((l[0] == ("1", 0)) and (l[1] == ("3", 0))) or
                     ((l[0] == ("3", 0)) and (l[1] == ("1", 0))))
     self.assertEqual(l[2], ("4", 2))
     self.assertEqual(l[3], ("2", 20))