def test_create_combination_list_2(self): calculator = Calculator() _input = ["apple", "orange", "grape", "banana"] _output = [["apple", "orange", "grape"], ["apple", "grape", "banana"], ["orange", "grape", "banana"]] self.assertListEqual(calculator.create_combination_list(_input, 3), _output)
if __name__ == "__main__": data = {} # データ要素を取り出す for table in TABLE_NAMES: columns = db_api.get_column_names(table) for column in columns: _list = db_api.pick_column(table, column) if column in ab_normal_column: _list = convert_abnormal_column_list(_list) data[f"{column}_sum"] = caluculator.create_sum_list(_list) data[f"{column}_average"] = caluculator.create_average_list( _list) elif column == "date": _list = [x.weekday() for x in _list] data["weekday"] = _list elif column in skip_column: pass else: _list = cast_list_float(_list) data[column] = _list keys = list(data.keys()) combine_list = caluculator.create_combination_list(keys, 2) for combi in combine_list: try: grapher.scatter_plot_with_corrcoef(data[combi[0]], combi[0], data[combi[1]], combi[1]) except BaseException: print(combi)