Skip to content

Commit b64cf11

Browse files
author
Alison Gim
committed
Update tests for intermediary measures without paren depth tracking
1 parent 323c257 commit b64cf11

1 file changed

Lines changed: 8 additions & 7 deletions

File tree

tests/unit/test_dialect_kql.py

Lines changed: 8 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -429,7 +429,8 @@ def test_select_count():
429429
'let inner_qry = (["logs"]);'
430430
"inner_qry"
431431
"| where Field1 > 1 and Field2 < 2"
432-
'| summarize ["total-count"] = count() '
432+
'| summarize ["__total-count_1"] = count() '
433+
'| extend ["total-count"] = ["__total-count_1"]'
433434
'| project ["total-count"]'
434435
'| order by ["total-count"] desc'
435436
"| take 5"
@@ -640,8 +641,8 @@ def test_calculated_measure_with_adhoc_measure_and_constant():
640641
).replace("\n", "")
641642
query_expected = (
642643
'["SalesData"]'
643-
'| summarize ["Measure 1"] = count() '
644-
'| extend ["Measure 2"] = ["Measure 1"] * 2'
644+
'| summarize ["__Measure 1_1"] = count() '
645+
'| extend ["Measure 1"] = ["__Measure 1_1"], ["Measure 2"] = ["Measure 1"] * 2'
645646
'| project ["Measure 1"], ["Measure 2"]'
646647
)
647648
assert query_compiled == query_expected
@@ -662,8 +663,8 @@ def test_calculated_measure_with_two_adhoc_measures_and_aggregates():
662663
).replace("\n", "")
663664
query_expected = (
664665
'["SalesData"]'
665-
'| summarize ["Measure 1"] = count(), ["Measure 2"] = count() '
666-
'| extend ["Measure 3"] = ["Measure 1"] + ["Measure 2"]'
666+
'| summarize ["__Measure 1_1"] = count() '
667+
'| extend ["Measure 1"] = ["__Measure 1_1"], ["Measure 2"] = ["__Measure 1_1"], ["Measure 3"] = ["Measure 1"] + ["Measure 2"]'
667668
'| project ["Measure 1"], ["Measure 2"], ["Measure 3"]'
668669
)
669670
assert query_compiled == query_expected
@@ -708,8 +709,8 @@ def test_calculated_measure_with_mixed_aggregates_and_references():
708709
).replace("\n", "")
709710
query_expected = (
710711
'["SalesData"]'
711-
'| summarize ["Predefined 1"] = count(), ["__Calculated_1"] = count(["b"]) '
712-
'| extend ["Calculated"] = ["Predefined 1"] + ["__Calculated_1"]'
712+
'| summarize ["__Predefined 1_1"] = count(), ["__Calculated_1"] = count(["b"]) '
713+
'| extend ["Predefined 1"] = ["__Predefined 1_1"], ["Calculated"] = ["Predefined 1"] + ["__Calculated_1"]'
713714
'| project ["Predefined 1"], ["Calculated"]'
714715
)
715716
assert query_compiled == query_expected

0 commit comments

Comments
 (0)