def test_iff_expression_return_type(self):
self.assertEqual(
" | project foo = iff(dateField > (ago(time(2.0:0:0.0))), array_length(arrayField), array_length(arrayField2))",
Query().project(foo=f.iff(t.dateField > f.ago(timedelta(2)), f.array_length(t.arrayField), f.array_length(t.arrayField2))).render()
)
def test_iff_related_types(self):
self.assertEqual(
" | project foo = iff(dateField > (ago(time(2.0:0:0.0))), 2, array_length(arrayField))",
Query().project(foo=f.iff(t.dateField > f.ago(timedelta(2)), 2, f.array_length(t.arrayField))).render()
)
def test_array_length(self):
self.assertEqual(
" | where (array_length(arrayField)) > 4",
Query().where(f.array_length(t.arrayField) > 4).render()
)
