def test_determine_commencement_period(self): pr = ProcessRoyalties() self.assertEqual(pr.determine_commencement_period(201501, date(2014, 12, 1)), 0.08) self.assertEqual(pr.determine_commencement_period(201501, date(2014, 12, 31)), 0) self.assertEqual(pr.determine_commencement_period(201501, date(2014, 1, 1)), 1) self.assertEqual(pr.determine_commencement_period(201501, date(2010, 11, 30)), 4.09) self.assertEqual(pr.determine_commencement_period(201501, date(2010, 1, 1)), 5) self.assertEqual(pr.determine_commencement_period(201501, date(2010, 1, 31)), 4.92) self.assertEqual(pr.determine_commencement_period(201501, date(2010, 1, 1)), 5.0) self.assertEqual(pr.determine_commencement_period(None, None), 5) self.assertEqual(pr.determine_commencement_period(201501, datetime(2003, 1, 1)), 12.01) return
def test_determineCommencementPeriod(self): pr = ProcessRoyalties() self.assertEqual(pr.determine_commencement_period(201501, datetime(2015, 1, 1)), 0) self.assertEqual(pr.determine_commencement_period(201501, datetime(2014, 12, 1)), 0.08) self.assertEqual(pr.determine_commencement_period(201501, datetime(2014, 11, 15)), 0.13) self.assertEqual(pr.determine_commencement_period(201501, datetime(2014, 1, 1)), 1) self.assertEqual(pr.determine_commencement_period(201501, datetime(2014, 1, 1)), 1) self.assertEqual(pr.determine_commencement_period(201501, datetime(2010, 1, 1)), 5) self.assertEqual(pr.determine_commencement_period(201501, datetime(2009, 12, 1)), 5.09) self.assertEqual(pr.determine_commencement_period(201501, None), 5)