Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[Snowflake] test types for unit testing #896

Merged
merged 13 commits into from
Feb 9, 2024
9 changes: 9 additions & 0 deletions dbt/include/snowflake/macros/utils/cast.sql
Original file line number Diff line number Diff line change
@@ -0,0 +1,9 @@
{% macro snowflake__cast(field, type) %}
{% if (type|upper == "GEOGRAPHY") -%}
to_geography({{field}})
{% elif (type|upper == "GEOMETRY") -%}
to_geometry({{field}})
{% else -%}
cast({{field}} as {{type}})
{% endif -%}
{% endmacro %}
mikealfare marked this conversation as resolved.
Show resolved Hide resolved
12 changes: 11 additions & 1 deletion dbt/include/snowflake/macros/utils/safe_cast.sql
Original file line number Diff line number Diff line change
@@ -1,3 +1,13 @@
{% macro snowflake__safe_cast(field, type) %}
try_cast({{field}} as {{type}})
{% if type|upper == "GEOMETRY" -%}
try_to_geometry({{field}})
{% elif type|upper == "GEOGRAPHY" -%}
try_to_geography({{field}})
{% elif type|upper != "VARIANT" -%}
mikealfare marked this conversation as resolved.
Show resolved Hide resolved
{#-- Snowflake try_cast does not support casting to variant, and expects the field as a string --#}
{% set field_as_sting = dbt.string_literal(field) if field is number else field %}
try_cast({{field_as_sting}} as {{type}})
MichelleArk marked this conversation as resolved.
Show resolved Hide resolved
{% else -%}
{{ adapter.dispatch('cast', 'dbt')(field, type) }}
{% endif -%}
{% endmacro %}
3 changes: 2 additions & 1 deletion dev-requirements.txt
Original file line number Diff line number Diff line change
@@ -1,7 +1,8 @@
# install latest changes in dbt-core
# TODO: how to automate switching from develop to version branches?
git+https://github.com/dbt-labs/dbt-core.git#egg=dbt-core&subdirectory=core
git+https://github.com/dbt-labs/dbt-core.git#egg=dbt-tests-adapter&subdirectory=tests/adapter
git+https://github.com/dbt-labs/dbt-adapters.git@unit-testing-case-insensitive-comparisons
MichelleArk marked this conversation as resolved.
Show resolved Hide resolved
git+https://github.com/dbt-labs/dbt-adapters.git@unit-testing-case-insensitive-comparisons#subdirectory=dbt-tests-adapter

# if version 1.x or greater -> pin to major version
# if version 0.x -> pin to minor
Expand Down
29 changes: 29 additions & 0 deletions tests/functional/adapter/unit_testing/test_unit_testing.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,29 @@
import pytest

from dbt.tests.adapter.unit_testing.test_unit_testing import BaseUnitTestingTypes


class TestSnowflakeUnitTestingTypes(BaseUnitTestingTypes):
@pytest.fixture
def data_types(self):
# sql_value, yaml_value
return [
["1", "1"],
["2.0", "2.0"],
["'12345'", "12345"],
["'string'", "string"],
["true", "true"],
["DATE '2020-01-02'", "2020-01-02"],
["TIMESTAMP '2013-11-03 00:00:00-0'", "2013-11-03 00:00:00-0"],
["'2013-11-03 00:00:00-0'::TIMESTAMPTZ", "2013-11-03 00:00:00-0"],
["TO_NUMBER('3', 10, 9)", "3"],
["3::VARIANT", "3"],
["TO_GEOMETRY('POINT(1820.12 890.56)')", "POINT(1820.12 890.56)"],
["TO_GEOGRAPHY('POINT(-122.35 37.55)')", "POINT(-122.35 37.55)"],
[
"{'Alberta':'Edmonton','Manitoba':'Winnipeg'}",
"{'Alberta':'Edmonton','Manitoba':'Winnipeg'}",
],
["['a','b','c']", "['a','b','c']"],
["[1,2,3]", "[1, 2, 3]"],
]
Loading