def test_read_dbf(self): df = read_sinan_dbf('test_data/EPR-2016-06-01-2016.dbf', encoding='latin-1') self.assertIsInstance(df, pd.DataFrame) for cname in df.columns: if cname.startswith('DT_'): self.assertIsInstance(df[cname][0], datetime.date) elif cname.startswith('SEM'): self.assertLessEqual(df[cname][0], 52) self.assertIsInstance(df[cname][0], (int, np.int64)) elif cname.startswith(('NU', 'ID')): if cname == 'ID_AGRAVO': continue self.assertIsInstance(df[cname][0], (int, float, np.int64), msg="Failed on column {}, type:{}".format(cname, type(df[cname][0])))
def test_read_dbf(self): df = read_sinan_dbf("test_data/EPR-2016-06-01-2016.dbf", encoding="latin-1") self.assertIsInstance(df, pd.DataFrame) for cname in df.columns: if cname.startswith("DT_"): self.assertIsInstance(df[cname][0], datetime.date) elif cname.startswith("SEM"): self.assertLessEqual(df[cname][0], 52) self.assertIsInstance(df[cname][0], (int, np.int64)) elif cname.startswith(("NU", "ID")): if cname == "ID_AGRAVO": continue self.assertIsInstance( df[cname][0], (int, float, np.int64), msg="Failed on column {}, type:{}".format( cname, type(df[cname][0])), )
def test_type_convertion(self): df = read_sinan_dbf('test_data/EPR-2016-06-01-2016.dbf', encoding='latin-1') assert not all(df.dtypes == 'object')
def test_type_convertion(self): df = read_sinan_dbf("test_data/EPR-2016-06-01-2016.dbf", encoding="latin-1") assert not all(df.dtypes == "object")