def test_next_workday(self): assert next_workday(self.sa) == self.mo assert next_workday(self.su) == self.mo assert next_workday(self.mo) == self.tu
def test_next_workday(day, expected): assert next_workday(day) == expected
day=8, observance=nearest_workday, ) NauryzHoliday1 = Holiday( "Nauryz Holiday", month=3, day=21, observance=next_monday, ) NauryzHoliday2 = Holiday( "Nauryz Holiday", month=3, day=21, observance=lambda dt: next_workday(next_monday(dt)), ) NauryzHoliday3 = Holiday( "Nauryz Holiday", month=3, day=21, observance=lambda dt: next_workday(next_workday(next_monday(dt))), ) KazakhstanPeopleSolidarityDay = Holiday( "Kazakhstan People Solidarity Day", month=5, day=1, observance=next_monday, )
def test_next_workday(self): self.assertEqual(next_workday(self.sa), self.mo) self.assertEqual(next_workday(self.su), self.mo) self.assertEqual(next_workday(self.mo), self.tu)
def next_trading_day(self, date: dt.date) -> dt.date: next_day = holiday.next_workday(date) while not self.is_trading_day(next_day): next_day = holiday.next_workday(next_day) return next_day