def test_nearest_workday(self): assert nearest_workday(self.sa) == self.fr assert nearest_workday(self.su) == self.mo assert nearest_workday(self.mo) == self.mo
def test_nearest_workday(day, expected): assert nearest_workday(day) == expected
def test_nearest_workday(self): self.assertEqual(nearest_workday(self.sa), self.fr) self.assertEqual(nearest_workday(self.su), self.mo) self.assertEqual(nearest_workday(self.mo), self.mo)
def nearest_workday_after_2013(dt): """ Nearest workday starting in 2014. """ return nearest_workday(dt) if dt.year > 2013 else dt