Skip to content
This repository was archived by the owner on Apr 8, 2025. It is now read-only.

Commit fd9f4e0

Browse files
committed
Add default type mapping for timestamp without time zone
1 parent 9bab6f0 commit fd9f4e0

File tree

2 files changed

+8
-0
lines changed

2 files changed

+8
-0
lines changed

cr8/insert_fake_data.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -100,6 +100,8 @@ class DataFaker:
100100
f.date_time_between, start_date='-2y', end_date='now'),
101101
'timestamp with time zone': lambda f: partial(
102102
f.date_time_between, start_date='-2y', end_date='now'),
103+
'timestamp without time zone': lambda f: partial(
104+
f.date_time_between, start_date='-2y', end_date='now'),
103105
'string': operator.attrgetter('word'),
104106
'text': operator.attrgetter('word'),
105107
'boolean': operator.attrgetter('boolean'),

tests/test_insert_fake_data.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -60,6 +60,12 @@ def test_timestamp_with_time_zone_type_default(self):
6060
diff = datetime.datetime(2017, 11, 18, 19, 0, 0) - dt
6161
self.assertLessEqual(diff, datetime.timedelta(seconds=1))
6262

63+
def test_timestamp_without_time_zone_type_default(self):
64+
provider = self.f.provider_for_column('some_ts_column', 'timestamp without time zone')
65+
dt = provider()
66+
diff = datetime.datetime(2017, 11, 18, 19, 0, 0) - dt
67+
self.assertLessEqual(diff, datetime.timedelta(seconds=1))
68+
6369
def test_provider_from_mapping(self):
6470
mapping = {'x': ['random_int', [10, 20]]}
6571
provider = self.f.provider_from_mapping('x', mapping)

0 commit comments

Comments
 (0)