def setUp(self):
"""
Set up an empty bibmatrix and one filled with ten clusters of 10 elements each.
"""
self.bm = Bib_matrix('testname', storage_dir_override='/tmp/')
self.css = ClusterSet()
self.css.clusters = [ClusterSet.Cluster(range(i*10,i*10+10)) for i in range(10)]
self.css.update_bibs()
self.bmcs0 = Bib_matrix('testname2', self.css, storage_dir_override='/tmp/')
def test_udate_all_bibs(self):
c = ClusterSet()
c.clusters = self.clusters
c.update_bibs()
self.assertTrue(c.num_all_bibs == 50)
self.assertTrue(
sorted(list((c.all_bibs()))) == list(
chain.from_iterable(
range(i * 10, i * 10 + 5) for i in range(10))))
