def test_check(self, formula, correct):
try:
formula_obj = FullFormula(formula)
formula_obj.check(*FullFormula.explain(formula))
is_correct = True
except FormulaException:
is_correct = False
self.assertEquals(is_correct, correct)
def test_generatelist(self, formula, year, dates_list):
formula_obj = FullFormula(formula, year)
try:
formula_obj.generatelist()
d_list = formula_obj.dates_list
except FormulaException:
d_list = False
self.assertEquals(d_list, dates_list)
def test_week_filter(self, w_filter, dates_list, result_dates_list):
# инициируем объект FullFormula с фильтром.
# сама формула не важна, потому как список дат мы ниже внедрим
formula_obj = FullFormula("xxx")
# внедрим список дат
formula_obj.dates_list = dates_list
formula_obj.week_filter(w_filter)
self.assertEquals(
result_dates_list,
formula_obj.dates_list,
"для фильтра {0} результат не верный. Получился {1}".format(w_filter, formula_obj.dates_list),
)
def test_explain(self, full_formula, w_filter, d_filter, formula):
frm, w_f, d_f = FullFormula.explain(full_formula)
self.assertEquals(
frm,
formula,
"в полной формуле {0} формула " "должна выйти {1}, но не {2}".format(full_formula, formula, frm),
)
self.assertEquals(
w_f,
w_filter,
"в полной формуле {0} фильтр дней " "должен выйти {1}, но не {2}".format(full_formula, w_filter, w_f),
)
self.assertEquals(
d_f,
d_filter,
"в полной формуле {0} фильтр данных " "должен выйти {1}, но не {2}".format(full_formula, d_filter, d_f),
)
def test_data_filter(self, d_filter, dates_list, result_dates_list):
# аналогично test_week_filter
formula_obj = FullFormula("xxx")
formula_obj.dates_list = dates_list
formula_obj.data_filter(d_filter)
self.assertEquals(result_dates_list, formula_obj.dates_list)
