QT/superset2
1
0
Fork 0
mirror of https://github.com/apache/superset.git synced 2026-04-19 16:14:52 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(KustoKQL): Update KQL alchemy version and update timegrain expressions (#32509)

Browse Source
This commit is contained in:
Ramachandran A G
2025-03-05 17:53:47 -08:00
committed by GitHub
parent c90e45a373
commit 317532752c
3 changed files with 35 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
tests/unit_tests/db_engine_specs/test_kusto.py
View File

@@ -19,7 +19,9 @@ from datetime import datetime
from typing import Optional
import pytest
from sqlalchemy import column
from superset.db_engine_specs.kusto import KustoKqlEngineSpec
from superset.sql.parse import SQLScript
from superset.sql_parse import ParsedQuery
from tests.unit_tests.db_engine_specs.utils import assert_convert_dttm
@@ -149,3 +151,25 @@ def test_sql_convert_dttm(
from superset.db_engine_specs.kusto import KustoSqlEngineSpec as spec # noqa: N813
assert_convert_dttm(spec, target_type, expected_result, dttm)
@pytest.mark.parametrize(
"in_duration,expected_result",
[
("PT1S", "bin(temporal,1s)"),
("PT1M", "bin(temporal,1m)"),
("PT5M", "bin(temporal,5m)"),
("PT1H", "bin(temporal,1h)"),
("P1D", "startofday(temporal)"),
("P1W", "startofweek(temporal)"),
("P1M", "startofmonth(temporal)"),
("P1Y", "startofyear(temporal)"),
],
)
def test_timegrain_expressions(in_duration: str, expected_result: str) -> None:
col = column("temporal")
actual_result = KustoKqlEngineSpec.get_timestamp_expr(
col=col, pdf=None, time_grain=in_duration
)
assert str(actual_result) == expected_result