def schema(self):
schema_dict = {}
if self.record:
schema_dict = self.record.get(self.SCHEMA)
return Schema.safe_init(schema_dict)
def schema(self):
schema_dict = {}
if self.record:
schema_dict = self.record.get(self.SCHEMA)
return Schema.safe_init(schema_dict)
def test_recognize_dates_from_schema(self):
schema = Schema({
'submit_date': {
SIMPLETYPE: DATETIME
}
})
df_with_dates = recognize_dates(self.dframe, schema)
for field in df_with_dates['submit_date']:
self.assertTrue(isinstance(field, datetime))
def test_rebuild_no_merge(self):
col = 'not-in-dframe'
schema = Schema({col: {}})
new_schema = schema.rebuild(self.dframe, overwrite=True)
self.assertFalse(col in new_schema)
def test_rebuild_merge(self):
col = 'not-in-dframe'
schema = Schema({col: {}})
new_schema = schema.rebuild(self.dframe)
self.assertEqual(new_schema[col], {})
def test_rebuild(self):
schema = Schema()
new_schema = schema.rebuild(self.dframe)
self.assertNotEqual(schema, new_schema)
def test_init(self):
schema = Schema()
self.assertTrue(isinstance(schema, dict))
