from freezegun import freeze_time from app.dao.date_util import get_current_calendar_year_start_year # see get_financial_year for conversion of financial years. @freeze_time("2017-03-31 23:59:59.999999") def test_get_current_calendar_year_start_year_before_march(): current_fy = get_current_calendar_year_start_year() assert current_fy == 2017 @freeze_time("2017-04-01 04:00:00.000000") def test_get_current_calendar_year_start_year_after_april(): current_fy = get_current_calendar_year_start_year() assert current_fy == 2017