| 480 | print "pattern.vector.Model.neighbors()" |
| 481 | |
| 482 | def test_search(self): |
| 483 | # Assert document vector space search. |
| 484 | v1 = self.model.search(self.model[0]) |
| 485 | v2 = self.model.search(vector.Document("cats meow")) |
| 486 | v3 = self.model.search(vector.Document("????")) |
| 487 | v4 = self.model.search("meow") |
| 488 | v5 = self.model.search(["cats", "meow"]) |
| 489 | self.assertEqual(v1, self.model.neighbors(self.model[0])) |
| 490 | self.assertEqual(v2[0][1], self.model[1]) |
| 491 | self.assertEqual(v3, []) |
| 492 | self.assertEqual(v4[0][1], self.model[1]) |
| 493 | self.assertEqual(v5[0][1], self.model[1]) |
| 494 | self.assertAlmostEqual(v4[0][0], 0.89, places=2) |
| 495 | self.assertAlmostEqual(v5[0][0], 1.00, places=2) |
| 496 | print "pattern.vector.Model.search()" |
| 497 | |
| 498 | def test_distance(self): |
| 499 | # Assert Model document distance. |