Example #1
0
    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)