def test_get_proportional_size_representatives(self): clusters = [ ClusterMock(range(0, 10)), ClusterMock(range(10, 50)), ClusterMock(range(50, 80)), ClusterMock(range(80, 200)) ] clustering = Clustering(clusters) rep = clustering.get_proportional_size_representatives( 30, "distance_matrix") self.assertItemsEqual(rep, [ 0, 0, 10, 10, 11, 12, 13, 14, 50, 50, 51, 52, 53, 80, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96 ])
def test_get_proportional_size_representatives(self): clusters = [ClusterMock(range(0,10)),ClusterMock(range(10,50)),ClusterMock(range(50,80)),ClusterMock(range(80,200))] clustering = Clustering(clusters) rep = clustering.get_proportional_size_representatives(30, "distance_matrix" ) self.assertItemsEqual(rep, [0, 0, 10, 10, 11, 12, 13, 14, 50, 50, 51, 52, 53, 80, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96])