def testPlanRevExceptionEmptyRanks(self):
dailyRevs = [5.0]
ranks = []
with self.assertRaises(ValueError):
rp = RevPlanner(dailyRevs, ranks)
def testPlanRevTwoRevOneEmptyRank(self):
dailyRevs = [5, 20]
ranks = [1]
rp = RevPlanner(dailyRevs, ranks)
self.assertEqual(rp.planRev(), [5.0])
def testPlanRevNormalDataBis(self):
dailyRevs = [5, 20]
ranks = [6, 2]
rp = RevPlanner(dailyRevs, ranks)
self.assertEqual(rp.planRev(), [25.0, 25.0, 5.0, 5.0, 5.0, 5.0])
def testPlanRevTwoRevSameRank(self):
dailyRevs = [5, 20]
ranks = [1, 1]
rp = RevPlanner(dailyRevs, ranks)
self.assertEqual(rp.planRev(), [25.0])