def test_after_nearest_workday(self): assert after_nearest_workday(self.sa) == self.mo assert after_nearest_workday(self.su) == self.tu assert after_nearest_workday(self.fr) == self.mo
def test_after_nearest_workday(day, expected): assert after_nearest_workday(day) == expected
def test_after_nearest_workday(self): self.assertEqual(after_nearest_workday(self.sa), self.mo) self.assertEqual(after_nearest_workday(self.su), self.tu) self.assertEqual(after_nearest_workday(self.fr), self.mo)