def execute_test(test_input, expected):
tel_dialect = TaxonTelDialect()
context = TelRootContext(SNOWFLAKE_HUSKY_CONTEXT, tel_dialect, None,
TAXON_MAP, node_id_maker())
result = tel_dialect.visit(test_input, context)
assert expected == result.literal_value(context)
def test_detection(expression, expected):
tel_dialect = TaxonTelDialect()
context = TelRootContext(SNOWFLAKE_HUSKY_CONTEXT,
tel_dialect,
None, {},
next_node_id=node_id_maker())
node = tel_dialect.visit(expression, context, skip_root_node=True)
res = can_become_comparison_metric(node)
assert res == expected