def test_compute_query_dict(self, x_range, y_range, query): bnac = BaseNonAggregate() bnac.chart_type = "test" bnac.x = "a" bnac.y = "b" bnac.x_range = x_range bnac.y_range = y_range df = cudf.DataFrame({"a": [1, 2, 2], "b": [3, 4, 5]}) dashboard = DashBoard(data=df) bnac.compute_query_dict(dashboard._query_str_dict) assert dashboard._query_str_dict["a_test"] == query
def test_compute_query_dict(self, x_range, y_range, query, local_dict): bnac = BaseNonAggregate() bnac.chart_type = "test" bnac.x = "x" bnac.y = "y" bnac.x_range = x_range bnac.y_range = y_range df = cudf.DataFrame({"x": [1, 2, 2], "y": [3, 4, 5]}) dashboard = DashBoard(dataframe=DataFrame.from_dataframe(df)) bnac.compute_query_dict(dashboard._query_str_dict, dashboard._query_local_variables_dict) assert dashboard._query_str_dict["x_test"] == query for key in local_dict: assert ( dashboard._query_local_variables_dict[key] == local_dict[key])
def test_compute_query_dict(self, x_range, y_range, query, local_dict): bnac = BaseNonAggregate() bnac.chart_type = "test" bnac.x = "x" bnac.y = "y" bnac.x_range = x_range bnac.y_range = y_range df = cudf.DataFrame({"x": [1, 2, 2], "y": [3, 4, 5]}) dashboard = DashBoard(dataframe=DataFrame.from_dataframe(df)) bnac.compute_query_dict(dashboard._query_str_dict, dashboard._query_local_variables_dict) bnac_key = (f"{bnac.x}_{bnac.y}" f"{'_' + bnac.aggregate_col if bnac.aggregate_col else ''}" f"_{bnac.aggregate_fn}_{bnac.chart_type}_{bnac.title}") assert dashboard._query_str_dict[bnac_key] == query for key in local_dict: assert ( dashboard._query_local_variables_dict[key] == local_dict[key])