def test_get_representative_vectors_from_empty_cluster(self): """ Test that when getting the representative vectors from an empty cluster, an empty list is returned. """ c = Cluster() self.assertEqual(list, type(c.get_representative_vectors(2))) self.assertEqual([], c.get_representative_vectors(2))
def test_get_representative_vectors(self): """ Test ranking the vectors according to their similarity to the cluster. """ v = [ Document("", ['a', 'b', 'c'], scheme=TF()), Document("", ['a', 'a', 'c'], scheme=TF()), Document("", ['p'], scheme=TF()), ] c = Cluster(v) self.assertEqual(list, type(c.get_representative_vectors(2))) self.assertEqual([v[1], v[0]], c.get_representative_vectors(2))
def test_get_representative_vector_from_empty_cluster(self): """ Test that when getting the representative vector from an empty cluster, ``None`` is returned. """ c = Cluster() self.assertEqual(None, c.get_representative_vectors(1))