Beispiel #1
0
 def test_filterClustersByAttribute(self):
     self.cluster1.addDocument(self.doc1)
     self.cluster2.addDocument(self.doc2)
     self.assertEqual([self.cluster1, self.cluster2],
                      list(
                          Cluster.getClustersByAttributeAndThreshold(
                              [self.cluster1, self.cluster2],
                              'vectorWeights', 1)))
     self.assertEqual([],
                      list(
                          Cluster.getClustersByAttributeAndThreshold(
                              [self.cluster1, self.cluster2],
                              'vectorWeights', 3)))
     self.assertEqual([self.cluster1, self.cluster2],
                      list(
                          Cluster.getClustersByAttributeAndThreshold(
                              [self.cluster1, self.cluster2],
                              'vectorWeights', 3, Cluster.BELOW_THRESHOLD)))
 def test_filterClustersByAttribute(self):
     self.cluster1.addDocument(self.doc1)
     self.cluster2.addDocument(self.doc2)
     self.assertEqual([self.cluster1, self.cluster2], list(Cluster.getClustersByAttributeAndThreshold([self.cluster1, self.cluster2], 'vectorWeights', 1)))
     self.assertEqual([], list(Cluster.getClustersByAttributeAndThreshold([self.cluster1, self.cluster2], 'vectorWeights', 3)))
     self.assertEqual([self.cluster1, self.cluster2], list(Cluster.getClustersByAttributeAndThreshold([self.cluster1, self.cluster2], 'vectorWeights', 3, Cluster.BELOW_THRESHOLD)))