def run_case_with_parameters(self,
case_num,
specific_params=None,
special_case=None):
"""Run a test case with parameters using ParameterDiscovery strategy.
Args:
case_num (int): The test case number to run.
"""
test_wb = load_original_case(case_num)
params = load_parameters_case(case_num)
params["data_ends"] = None
exp_dfs = load_expected_case(case_num, special_case)
if specific_params:
for specific_param, value in specific_params.items():
params[specific_param] = value
# get dfs from the strategy
series = XlSeries(test_wb)
test_dfs = series.get_data_frames(params)
if type(test_dfs) != list:
test_dfs = [test_dfs]
if type(exp_dfs) != list:
exp_dfs = [exp_dfs]
for test_df, exp_df in zip(test_dfs, exp_dfs):
print(test_df.columns, exp_df.columns)
self.assertTrue(compare_data_frames(test_df, exp_df))
def test_get_period_ranges(self):
test_wb = load_original_case(2)
params = load_parameters_case(2)
strategy_obj = ParameterDiscovery(test_wb, params)
ws = strategy_obj.wb.active
pr_d = pd.date_range("20020304", "20140410", freq="D")
pr_m = pd.date_range("20020301", "20140301", freq="MS")
period_ranges = list(strategy_obj._get_period_ranges(ws, params))
self.assertTrue(compare_period_ranges(pr_d, period_ranges[0]))
self.assertTrue(compare_period_ranges(pr_m, period_ranges[1]))
def test_make_ws_copy(self):
wb = load_original_case(2)
ws = wb.active
ws_copy = make_ws_copy(ws)
self.assertTrue(compare_cells_ws(ws, ws_copy))
def test_make_wb_copy(self):
wb = load_original_case(2)
wb_copy = make_wb_copy(wb)
self.assertTrue(compare_cells(wb, wb_copy))