Skip to content

Commit 3333ca0

Browse files
author
Jesse Whitehouse
committed
Update tests to check for inferred dates.
Only test_parameterized_query_named_dict_and_not_inferred_e2e passes because we don't infer anything here. Signed-off-by: Jesse Whitehouse <jesse.whitehouse@databricks.com>
1 parent b5e0726 commit 3333ca0

1 file changed

Lines changed: 5 additions & 5 deletions

File tree

tests/e2e/common/parameterized_query_tests.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,7 @@ def test_parameterized_query_named_and_inferred_e2e(self):
3131
"p_bool": True,
3232
"p_int": 1234,
3333
"p_double": 3.14,
34-
"p_date": datetime(2023, 9, 6, 3, 14, 27, 843, tzinfo=pytz.UTC),
34+
"p_date": datetime(2023, 9, 6, 3, 14, 27, 843, tzinfo=pytz.UTC).date(),
3535
"p_timestamp": datetime(2023, 9, 6, 3, 14, 27, 843, tzinfo=pytz.UTC),
3636
"p_str": "Hello",
3737
}
@@ -50,7 +50,7 @@ def test_parameterized_query_named_and_inferred_e2e(self):
5050

5151
assert result.col_date == datetime(
5252
2023, 9, 6, 3, 14, 27, 843, tzinfo=pytz.UTC
53-
)
53+
).date()
5454
assert result.col_timestamp == datetime(
5555
2023, 9, 6, 3, 14, 27, 843, tzinfo=pytz.UTC
5656
)
@@ -88,7 +88,7 @@ def test_parameterized_query_named_dict_and_inferred_e2e(self):
8888
),
8989
DbSqlParameter(
9090
name="p_date",
91-
value=datetime(2023, 9, 6, 3, 14, 27, 843, tzinfo=pytz.UTC),
91+
value=datetime(2023, 9, 6, 3, 14, 27, 843, tzinfo=pytz.UTC).date(),
9292
),
9393
DbSqlParameter(
9494
name="p_timestamp",
@@ -112,7 +112,7 @@ def test_parameterized_query_named_dict_and_inferred_e2e(self):
112112

113113
assert result.col_date == datetime(
114114
2023, 9, 6, 3, 14, 27, 843, tzinfo=pytz.UTC
115-
)
115+
).date()
116116
assert result.col_timestamp == datetime(
117117
2023, 9, 6, 3, 14, 27, 843, tzinfo=pytz.UTC
118118
)
@@ -141,7 +141,7 @@ def test_parameterized_query_named_dict_and_not_inferred_e2e(self):
141141
DbSqlParameter(name="p_double", value=3.14, type=DbSqlType.FLOAT),
142142
DbSqlParameter(
143143
name="p_date",
144-
value=datetime(2023, 9, 6, 3, 14, 27, 843, tzinfo=pytz.UTC),
144+
value=datetime(2023, 9, 6, 3, 14, 27, 843, tzinfo=pytz.UTC).date(),
145145
type=DbSqlType.DATE,
146146
),
147147
DbSqlParameter(

0 commit comments

Comments
 (0)