def test_next_monday_or_tuesday(self): assert next_monday_or_tuesday(self.sa) == self.mo assert next_monday_or_tuesday(self.su) == self.tu assert next_monday_or_tuesday(self.mo) == self.tu
def test_next_monday_or_tuesday(day, expected): assert next_monday_or_tuesday(day) == expected
def test_next_monday_or_tuesday(self): self.assertEqual(next_monday_or_tuesday(self.sa), self.mo) self.assertEqual(next_monday_or_tuesday(self.su), self.tu) self.assertEqual(next_monday_or_tuesday(self.mo), self.tu)
def new_years_day_observance(holidays): # There was no extra observance of New Year's Day in 2006. holidays = holidays[holidays.year != 2006] return pd.to_datetime([next_monday_or_tuesday(day) for day in holidays])