| 279 | print "pattern.vector.Document.load()" |
| 280 | |
| 281 | def test_document_vector(self): |
| 282 | # Assert Vector properties. |
| 283 | # Test copy. |
| 284 | v = vector.Document("the cat sat on the mat").vector |
| 285 | v = v.copy() |
| 286 | # Test properties. |
| 287 | self.assertTrue(isinstance(v, dict)) |
| 288 | self.assertTrue(isinstance(v, vector.Vector)) |
| 289 | self.assertTrue(isinstance(v.id, int)) |
| 290 | self.assertEqual(sorted(v.features), ["cat", "mat", "sat"]) |
| 291 | self.assertEqual(v.weight, vector.TF) |
| 292 | self.assertAlmostEqual(v.norm, 0.58, places=2) |
| 293 | self.assertAlmostEqual(v["cat"], 0.33, places=2) |
| 294 | self.assertAlmostEqual(v["sat"], 0.33, places=2) |
| 295 | self.assertAlmostEqual(v["mat"], 0.33, places=2) |
| 296 | # Test copy + update. |
| 297 | v = v({"cat":1, "sat":1, "mat":1}) |
| 298 | self.assertEqual(sorted(v.features), ["cat", "mat", "sat"]) |
| 299 | self.assertAlmostEqual(v["cat"], 1.00, places=2) |
| 300 | self.assertAlmostEqual(v["sat"], 1.00, places=2) |
| 301 | self.assertAlmostEqual(v["mat"], 1.00, places=2) |
| 302 | print "pattern.vector.Document.vector" |
| 303 | |
| 304 | def test_document_keywords(self): |
| 305 | # Assert Document.keywords() based on term frequency. |