def test_csv(self): filepath = p.join(io.DATA_DIR, 'test.csv') header = ['some_date', 'sparse_data', 'some_value', 'unicode_test'] with open(filepath, 'r', encoding='utf-8') as f: records = io._read_csv(f, header) nt.assert_equal(self.sheet0_alt, next(records)) filepath = p.join(io.DATA_DIR, 'no_header_row.csv') records = io.read_csv(filepath, has_header=False) expected = {'column_1': '1', 'column_2': '2', 'column_3': '3'} nt.assert_equal(expected, next(records)) filepath = p.join(io.DATA_DIR, 'test_bad.csv') kwargs = {'sanitize': True, 'first_row': 1, 'first_col': 1} records = io.read_csv(filepath, **kwargs) nt.assert_equal(self.sheet0_alt, next(records)) filepath = p.join(io.DATA_DIR, 'fixed_w_header.txt') widths = [0, 18, 29, 33, 38, 50] records = io.read_fixed_fmt(filepath, widths, has_header=True) expected = { 'News Paper': 'Chicago Reader', 'Founded': '1971-01-01', 'Int': '40', 'Bool': 'True', 'Float': '1.0', 'Timestamp': '04:14:001971-01-01T04:14:00' } nt.assert_equal(expected, next(records))
def test_csv(self): """Test for reading csv files""" filepath = p.join(io.DATA_DIR, 'no_header_row.csv') records = io.read_csv(filepath, has_header=False) expected = {'column_1': '1', 'column_2': '2', 'column_3': '3'} nt.assert_equal(expected, next(records)) filepath = p.join(io.DATA_DIR, 'test_bad.csv') kwargs = {'sanitize': True, 'first_row': 1, 'first_col': 1} records = io.read_csv(filepath, **kwargs) nt.assert_equal(self.sheet0_alt, next(records)) filepath = p.join(io.DATA_DIR, 'fixed_w_header.txt') widths = [0, 18, 29, 33, 38, 50] records = io.read_fixed_fmt(filepath, widths, has_header=True) expected = { 'News Paper': 'Chicago Reader', 'Founded': '1971-01-01', 'Int': '40', 'Bool': 'True', 'Float': '1.0', 'Timestamp': '04:14:001971-01-01T04:14:00' } nt.assert_equal(expected, next(records))
def test_csv(self): """Test for reading csv files""" filepath = p.join(io.DATA_DIR, "no_header_row.csv") records = io.read_csv(filepath, has_header=False) expected = {"column_1": "1", "column_2": "2", "column_3": "3"} nt.assert_equal(expected, next(records)) filepath = p.join(io.DATA_DIR, "test_bad.csv") kwargs = {"sanitize": True, "first_row": 1, "first_col": 1} records = io.read_csv(filepath, **kwargs) nt.assert_equal(self.sheet0_alt, next(records)) filepath = p.join(io.DATA_DIR, "fixed_w_header.txt") widths = [0, 18, 29, 33, 38, 50] records = io.read_fixed_fmt(filepath, widths, has_header=True) expected = { "News Paper": "Chicago Reader", "Founded": "1971-01-01", "Int": "40", "Bool": "True", "Float": "1.0", "Timestamp": "04:14:001971-01-01T04:14:00", } nt.assert_equal(expected, next(records))
def test_csv(self): filepath = p.join(io.DATA_DIR, 'test.csv') header = ['some_date', 'sparse_data', 'some_value', 'unicode_test'] with open(filepath, 'rU', encoding='utf-8') as f: records = io._read_csv(f, header) nt.assert_equal(self.sheet0_alt, next(records)) filepath = p.join(io.DATA_DIR, 'no_header_row.csv') records = io.read_csv(filepath, has_header=False) expected = {'column_1': '1', 'column_2': '2', 'column_3': '3'} nt.assert_equal(expected, next(records)) filepath = p.join(io.DATA_DIR, 'test_bad.csv') kwargs = {'sanitize': True, 'first_row': 1, 'first_col': 1} records = io.read_csv(filepath, **kwargs) nt.assert_equal(self.sheet0_alt, next(records)) filepath = p.join(io.DATA_DIR, 'fixed_w_header.txt') widths = [0, 18, 29, 33, 38, 50] records = io.read_fixed_fmt(filepath, widths, has_header=True) expected = { 'News Paper': 'Chicago Reader', 'Founded': '1971-01-01', 'Int': '40', 'Bool': 'True', 'Float': '1.0', 'Timestamp': '04:14:001971-01-01T04:14:00'} nt.assert_equal(expected, next(records))
def test_csv(self): """Test for reading csv files""" filepath = p.join(io.DATA_DIR, 'no_header_row.csv') records = io.read_csv(filepath, has_header=False) expected = {'column_1': '1', 'column_2': '2', 'column_3': '3'} nt.assert_equal(expected, next(records)) filepath = p.join(io.DATA_DIR, 'test_bad.csv') kwargs = {'sanitize': True, 'first_row': 1, 'first_col': 1} records = io.read_csv(filepath, **kwargs) nt.assert_equal(self.sheet0_alt, next(records)) filepath = p.join(io.DATA_DIR, 'fixed_w_header.txt') widths = [0, 18, 29, 33, 38, 50] records = io.read_fixed_fmt(filepath, widths, has_header=True) expected = { 'News Paper': 'Chicago Reader', 'Founded': '1971-01-01', 'Int': '40', 'Bool': 'True', 'Float': '1.0', 'Timestamp': '04:14:001971-01-01T04:14:00'} nt.assert_equal(expected, next(records))