def test_compute_query_dict(self, x_range, y_range, query, local_dict): bg = BaseGraph() bg.chart_type = "test" bg.x_range = x_range bg.y_range = y_range df = cudf.DataFrame({"x": [1, 2, 2], "y": [3, 4, 5]}) dashboard = DashBoard(dataframe=DataFrame.from_dataframe(df)) bg.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, df_type, x_range, y_range, query, local_dict ): bg = BaseGraph() bg.chart_type = "test" bg.box_selected_range = local_dict bg.x_range = x_range bg.y_range = y_range df = initialize_df(df_type, {"x": [1, 2, 2], "y": [3, 4, 5]}) dashboard = DashBoard(dataframe=DataFrame.from_dataframe(df)) bg.compute_query_dict( dashboard._query_str_dict, dashboard._query_local_variables_dict ) assert ( dashboard._query_str_dict[f"x_y_vertex_count_test_{bg.title}"] == query ) for key in local_dict: assert ( dashboard._query_local_variables_dict[key] == local_dict[key] )