def test_get_dtypes_given_header(self): """ test if dtypes with custom header can be returned """ spec = {'name': "nohead", 'type': "csv"} cols = [ "Lat", "Lon", "Txt", "Nam", "Add", "Job", ] g = FileInspectorCsvTest._gen_file(name=spec['name'], header=False, type=spec['type']) g.create() c = FileInspector(cfg_path=get_inipath(), file=os.path.join(FILEPATH, (f"{spec['name']}." f"{spec['type']}")), seperator="|", type="csv", columns=cols, header=None) dtype_key_list = c.get_dtypes().keys() g.rm() self.assertEqual(sorted(cols), sorted(dtype_key_list))
def test_get_dtypes_no_header(self): """ test if dtypes without header can be returned """ spec = {'name': "nohead", 'type': "csv"} g = FileInspectorCsvTest._gen_file(name=spec['name'], header=False, type=spec['type']) g.create() c = FileInspector(cfg_path=get_inipath(), file=os.path.join(FILEPATH, (f"{spec['name']}." f"{spec['type']}")), seperator="|", type="csv", header=None) with self.assertRaises(NoColumnsError): c.get_dtypes().keys() g.rm()
def test_get_dtypes(self): """ test if columns """ c = FileInspector( cfg_path=get_inipath(), file=os.path.join(FILEPATH, (f"{FileInspectorJsonTest.testfile['name']}." f"{FileInspectorJsonTest.testfile['type']}")), type="json", orient="split") dtype_key_list = c.get_dtypes().keys() exp_col = ["Lat", "Lon", "Txt", "Nam", "Add", "Job", "CreatedAt"] self.assertEqual(sorted(exp_col), sorted(dtype_key_list))
def test_change_dt_to_string(self): """ test if all columns can be changed to string datatype """ c = FileInspector( cfg_path=get_inipath(), file=os.path.join(FILEPATH, (f"{FileInspectorCsvTest.testfile['name']}." f"{FileInspectorCsvTest.testfile['type']}")), seperator="|", type="csv") c.col_to_str() dtype_key_list = c.get_dtypes().values() check = all(x.name == "object" for x in dtype_key_list) self.assertEqual(check, True)
def test_get_dtypes(self): """ test if data types gets returned as a dict """ c = FileInspector( cfg_path=get_inipath(), file=os.path.join(FILEPATH, (f"{FileInspectorCsvTest.testfile['name']}." f"{FileInspectorCsvTest.testfile['type']}")), seperator="|", type="csv") dtype_key_list = c.get_dtypes().keys() exp_col = [ "Lat", "Lon", "Txt", "Nam", "Add", "Job", ] self.assertEqual(sorted(exp_col), sorted(dtype_key_list))