add:names test of TimestampType

snowflakedb · Jan 25, 2025 · 713efd8 · 713efd8
1 parent 32c2f4a
commit 713efd8
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/unit/test_types.py b/tests/unit/test_types.py
@@ -233,6 +233,7 @@ def test_sf_datatype_names():
     assert str(FloatType()) == "FloatType()"
     assert str(DoubleType()) == "DoubleType()"
     assert str(DecimalType(1, 2)) == "DecimalType(1, 2)"
+    assert str(TimestampType(TimestampTimeZone.TZ)) == "TimestampType(timezone=TimestampTimeZone('tz'))"
 
 
 def test_sf_datatype_hashes():
@@ -258,6 +259,7 @@ def test_sf_datatype_hashes():
     assert hash(FloatType()) == hash("FloatType()")
     assert hash(DoubleType()) == hash("DoubleType()")
     assert hash(DecimalType(1, 2)) == hash("DecimalType(1, 2)")
+    assert hash(TimestampType(TimestampTimeZone.TZ)) == hash("TimestampType(timezone=TimestampTimeZone('tz'))")
 
 
 def test_merge_type():