def test_count(self):
self.assertEqual(
" | summarize count()",
Query().summarize(f.count()).render()
)
self.assertEqual(
" | summarize count(stringField)",
Query().summarize(f.count(t.stringField)).render()
)
def test_summarize_by(self):
self.assertEqual(
"mock_table | summarize count(stringField), my_count = count(stringField2) by boolField, bin(numField, 1), time_range = bin(dateField, 10)",
Query(t).summarize(f.count(t.stringField),
my_count=f.count(t.stringField2)).by(
t.boolField,
f.bin(t.numField, 1),
time_range=f.bin(t.dateField, 10)).render(),
)
def test_summarize_by_with_shuffle_hint_specific_key(self):
self.assertEqual(
"mock_table | summarize hint.shufflekey = boolField count(stringField) by boolField, time_range = bin(dateField, time(0.0:0:10.0))",
(Query(t).summarize(f.count(t.stringField)).by(
t.boolField,
time_range=f.bin(t.dateField, timedelta(
seconds=10))).shuffle_key(t.boolField).render()))
def test_summarize_by_with_shuffle_hint_multiple_specific_keys(self):
self.assertEqual(
"mock_table | summarize hint.shufflekey = boolField hint.shufflekey = stringField count(stringField) by boolField, stringField, numField",
(Query(t).summarize(f.count(t.stringField)).by(
t.boolField, t.stringField, t.numField).shuffle_key(
t.boolField, t.stringField).render()))
def test_summarize_by_with_shuffle_hint(self):
self.assertEqual(
"mock_table | summarize hint.strategy = shuffle count(stringField) by tostring(mapField)",
Query(t).summarize(f.count(t.stringField)).by(
f.to_string(t.mapField)).shuffle_key().render())
def test_summarize_by_expression(self):
self.assertEqual(
"mock_table | summarize count(stringField) by tostring(mapField)",
Query(t).summarize(f.count(t.stringField)).by(
f.to_string(t.mapField)).render(),
)
def test_summarize(self):
self.assertEqual(
"mock_table | summarize count(stringField), my_count = count(stringField2)",
Query(t).summarize(f.count(t.stringField),
my_count=f.count(t.stringField2)).render(),
)
