fix: TemporalWrapperType string representation (#16614)

* fix: TemporalWrapperType string representation

* fix tests

(cherry picked from commit 9de2196b7f)
This commit is contained in:
Ville Brofeldt
2021-09-07 13:50:24 +03:00
committed by Ville Brofeldt
parent 4a92ba0b12
commit 72fe94842d
4 changed files with 19 additions and 9 deletions

View File

@@ -561,11 +561,13 @@ class TestPrestoDbEngineSpec(TestDbEngineSpec):
self.assertEqual(column_spec.generic_type, GenericDataType.NUMERIC)
column_spec = PrestoEngineSpec.get_column_spec("time")
assert issubclass(column_spec.sqla_type, types.Time)
assert isinstance(column_spec.sqla_type, types.Time)
assert type(column_spec.sqla_type).__name__ == "TemporalWrapperType"
self.assertEqual(column_spec.generic_type, GenericDataType.TEMPORAL)
column_spec = PrestoEngineSpec.get_column_spec("timestamp")
assert issubclass(column_spec.sqla_type, types.TIMESTAMP)
assert isinstance(column_spec.sqla_type, types.TIMESTAMP)
assert type(column_spec.sqla_type).__name__ == "TemporalWrapperType"
self.assertEqual(column_spec.generic_type, GenericDataType.TEMPORAL)
sqla_type = PrestoEngineSpec.get_sqla_column_type(None)