class A_Calendar:
def setup_method(self, method):
self.calendar = Calendar()
self.start_date = [
datetime.date(2016, 2, 10),
datetime.date(2016, 2, 15),
datetime.date(2016, 2, 15),
datetime.date(2016, 2, 3)
]
self.end_date = [
datetime.date(2016, 2, 12),
datetime.date(2016, 2, 15),
datetime.date(2016, 2, 15),
datetime.date(2016, 2, 9)
]
self.duration = [3, 1, 0, 5]
def should_calculate_end_date(self):
for i in range(0, 4):
assert self.end_date[i] == self.calendar.get_end_date(
self.start_date[i], self.duration[i])
def not_yet_should_calculate_duration(self):
r = [0, 1, 3]
for i in r:
assert self.duration[i] == self.calendar.get_duration(
self.start_date[i], self.end_date[i])
def not_yet_should_calculate_start_date(self):
r = [0, 1, 3]
for i in r:
assert self.start_date[i] == self.calendar.get_start_date(
self.duration[i], self.end_date[i])
def setup_method(self, method):
self.calendar = Calendar()
self.start_date = [
datetime.date(2016, 2, 10),
datetime.date(2016, 2, 15),
datetime.date(2016, 2, 15),
datetime.date(2016, 2, 3)
]
self.end_date = [
datetime.date(2016, 2, 12),
datetime.date(2016, 2, 15),
datetime.date(2016, 2, 15),
datetime.date(2016, 2, 9)
]
self.duration = [3, 1, 0, 5]
