diff options
Diffstat (limited to 'python/tests')
| -rw-r--r-- | python/tests/test_dates.py | 27 |
1 files changed, 27 insertions, 0 deletions
diff --git a/python/tests/test_dates.py b/python/tests/test_dates.py index bebe4641..843941db 100644 --- a/python/tests/test_dates.py +++ b/python/tests/test_dates.py @@ -5,6 +5,7 @@ import pandas as pd import sys sys.path.append('..') from analytics.utils import * +from pyisda.date import roll_date as roll_date_c class TestStartDate(unittest.TestCase): @@ -52,6 +53,32 @@ class TestEndDate(unittest.TestCase): datetime.date(2020, 12, 20)) self.assertEqual(roll_date(datetime.date(2015, 12, 20), 5), datetime.date(2020, 12, 20)) + +class TestEndDateC(unittest.TestCase): + roll_date_c = lambda d, t: TDate_to_pydate(end_dates(pydate_to_TDate(d), t)) + + def test_enddate(self): + self.assertEqual(roll_date_c(datetime.date(2016, 9, 20), 5), + datetime.date(2021, 12, 20)) + self.assertEqual(roll_date_c(datetime.date(2016, 10, 15), 5), + datetime.date(2021, 12, 20)) + self.assertEqual(roll_date_c(datetime.date(2017, 3, 19), 5), + datetime.date(2021, 12, 20)) + self.assertEqual(roll_date_c(datetime.date(2017, 3, 20), 5), + datetime.date(2022, 6, 20)) + + def test_enddate_pre2015(self): + self.assertEqual(roll_date_c(datetime.date(2014, 9, 20), 5), + datetime.date(2019, 12, 20)) + self.assertEqual(roll_date_c(datetime.date(2014, 10, 15), 5), + datetime.date(2019, 12, 20)) + self.assertEqual(roll_date_c(datetime.date(2015, 3, 19), 5), + datetime.date(2020, 3, 20)) + self.assertEqual(roll_date_c(datetime.date(2015, 3, 20), 5), + datetime.date(2020, 6, 20)) + self.assertEqual(roll_date_c(datetime.date(2015, 12, 19), 5), + datetime.date(2020, 12, 20)) + self.assertEqual(roll_date_c(datetime.date(2015, 12, 20), 5), datetime.date(2020, 12, 20)) if __name__=="__main__": |