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))