def test_write_default_config(self): import tempfile tdir = tempfile.mkdtemp() outpth = os.path.join(tdir, 'defconfig.yml') SimpleDelimitedParser.write_default_parser_config(outpth) parser = SimpleDelimitedParser.get_parser( 'test_data/testmanualupload.csv', outpth) props = parser.properties want_props = { 'data_url': 'test_data/testmanualupload.csv', 'header_size': 0, 'delimiter': ',', 'num_columns': None, 'use_column_names': True, 'dtype': 'float32', 'fill_val': -999, 'column_map': None } for p in props.keys(): self.assertEqual(props[p], want_props[p]) import shutil shutil.rmtree(outpth, ignore_errors=True)
def test_write_default_config(self): import tempfile tdir = tempfile.mkdtemp() outpth = os.path.join(tdir, 'defconfig.yml') SimpleDelimitedParser.write_default_parser_config(outpth) parser = SimpleDelimitedParser.get_parser('test_data/testmanualupload.csv', outpth) props = parser.properties want_props = { 'data_url': 'test_data/testmanualupload.csv', 'header_size': 0, 'delimiter': ',', 'num_columns': None, 'use_column_names': True, 'dtype': 'float32', 'fill_val': -999, 'column_map': None } for p in props.keys(): self.assertEqual(props[p], want_props[p]) import shutil shutil.rmtree(outpth, ignore_errors=True)