Skip to content

Commit

Permalink
fix test
Browse files Browse the repository at this point in the history
  • Loading branch information
colin-sentry committed Feb 24, 2025
1 parent 2b4e946 commit a6d6240
Showing 1 changed file with 3 additions and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -220,7 +220,9 @@
@pytest.mark.parametrize("pre_format, expected_query", test_data)
def test_format_expressions(pre_format: Query, expected_query: Query) -> None:
copy = deepcopy(pre_format)
HashBucketFunctionTransformer("attr_str").process_query(copy, HTTPQuerySettings())
HashBucketFunctionTransformer("attr_str", num_attribute_buckets=5).process_query(
copy, HTTPQuerySettings()
)
assert copy.get_selected_columns() == expected_query.get_selected_columns()
assert copy.get_groupby() == expected_query.get_groupby()
assert copy.get_condition() == expected_query.get_condition()

0 comments on commit a6d6240

Please sign in to comment.