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]
)