def test_setting_vectors(self):
"""
Test setting the vectors manually.
"""
v = [
Document("", ["a", "b", "a", "c"], scheme=TF()),
Document("", ["a", "c"], scheme=TF())
]
c = Cluster()
self.assertEqual({}, c.centroid.dimensions)
c.vectors = v
self.assertEqual(v, c.vectors)
def test_set_vectors_none(self):
"""
Test that setting vectors to ``None`` overwrites existing vectors.
"""
v = [
Document("", ["a", "b", "a", "c"], scheme=TF()),
Document("", ["a", "c"], scheme=TF())
]
c = Cluster(v)
self.assertEqual(v, c.vectors)
c.vectors = None
self.assertEqual([], c.vectors)
self.assertEqual({}, c.centroid.dimensions)
def test_set_one_vectors(self):
"""
Test that setting vectors to a single vector overwrites existing vectors.
"""
v = [
Document("", ["a", "b", "a", "c"], scheme=TF()),
Document("", ["a", "c"], scheme=TF())
]
c = Cluster(v)
self.assertEqual(v, c.vectors)
n = Document("", ['a'], scheme=TF())
c.vectors = n
self.assertEqual([n], c.vectors)
self.assertEqual(n.dimensions, c.centroid.dimensions)
def test_set_several_vectors(self):
"""
Test that setting vectors to several vectors overwrites existing vectors.
"""
v = Document("", ['a'], scheme=TF())
c = Cluster(v)
self.assertEqual([v], c.vectors)
self.assertEqual(v.dimensions, c.centroid.dimensions)
n = [
Document("", ["a", "b", "a", "c"], scheme=TF()),
Document("", ["a", "c"], scheme=TF())
]
c.vectors = n
self.assertEqual(n, c.vectors)
