class TestRank(unittest.TestCase):
def setUp(self):
return
def test_1_insert_init(self):
self.vpicReader = VPICReader('.')
self.rank = Rank(self.vpicReader, 2)
self.rank.insert(range(20))
self.assertEqual(len(self.rank.oob_left), 20,
msg="init insertion not okay")
self.assertEqual(len(self.rank.oob_right), 0,
msg="init insertion not okay")
self.assertIsNone(self.rank.pivots)
self.assertIsNone(self.rank.pivot_counts)
def test_2_rank_pivots(self):
vpicReader = VPICReader('.')
rank = Rank(vpicReader, 2)
rank.insert(range(20))
pivots, pivot_width = rank.compute_pivots(5)
self.assertEqual(pivots, [0, 5, 10, 15, 19])
self.assertAlmostEqual(pivot_width, 5.0)
return
def test_3_update_pivots(self):
new_pivots = [3, 7, 11, 13]
self.rank.update_pivots(new_pivots)
self.rank.flush_oobs()
self.assertEqual(self.rank.pivot_counts, [4, 4, 2])
self.assertEqual(len(self.rank.oob_left), 3)
self.assertEqual(len(self.rank.oob_right), 7)
def test_4_get_pivots_again(self):
pivots, pivot_width = self.rank.compute_pivots(3)
self.assertEqual(pivots, [0, 10, 19])
self.assertAlmostEqual(pivot_width, 10.0)
pivots, pivot_width = self.rank.compute_pivots(7)
self.assertEqual(pivots, [0, 3, 7, 10, 13, 17, 19])
def test_repartition(self):
pivots_old = [3, 10, 13, 19]
counts_old = [2, 7, 3]
pivots_new = [2, 4, 10, 13, 23]
counts_new = []
self.rank.__class__.repartition_bin_counts(pivots_old, counts_old,
pivots_new, counts_new)
self.assertAlmostEqual(sum(counts_old), sum(counts_new))
self.assertAlmostEqual(counts_new[0], 0.2857, 3)
self.assertAlmostEqual(counts_new[3], 3.0, 3)
return
def tearDown(self):
return
def test_2_rank_pivots(self):
vpicReader = VPICReader('.')
rank = Rank(vpicReader, 2)
rank.insert(range(20))
pivots, pivot_width = rank.compute_pivots(5)
self.assertEqual(pivots, [0, 5, 10, 15, 19])
self.assertAlmostEqual(pivot_width, 5.0)
return
