def test_xls(self):
"""Test for reading excel files"""
filepath = p.join(io.DATA_DIR, 'test.xlsx')
records = io.read_xls(filepath, sanitize=True, sheet=0)
nt.assert_equal(self.sheet0, next(records))
records = io.read_xls(filepath, sanitize=True, sheet=1)
nt.assert_equal(self.sheet1, next(records))
kwargs = {'first_row': 1, 'first_col': 1}
records = io.read_xls(filepath, sanitize=True, sheet=2, **kwargs)
nt.assert_equal(self.sheet0, next(records))
records = io.read_xls(filepath, sanitize=True, sheet=3, **kwargs)
nt.assert_equal(self.sheet1, next(records))
def test_xls(self):
"""Test for reading excel files"""
filepath = p.join(io.DATA_DIR, "test.xlsx")
records = io.read_xls(filepath, sanitize=True, sheet=0)
nt.assert_equal(self.sheet0, next(records))
records = io.read_xls(filepath, sanitize=True, sheet=1)
nt.assert_equal(self.sheet1, next(records))
kwargs = {"first_row": 1, "first_col": 1}
records = io.read_xls(filepath, sanitize=True, sheet=2, **kwargs)
nt.assert_equal(self.sheet0, next(records))
records = io.read_xls(filepath, sanitize=True, sheet=3, **kwargs)
nt.assert_equal(self.sheet1, next(records))
def test_xls(self):
filepath = p.join(io.DATA_DIR, 'test.xlsx')
records = io.read_xls(filepath, sanitize=True, sheet=0)
nt.assert_equal(self.sheet0, next(records))
with open(filepath, 'r+b') as f:
records = io.read_xls(f, sanitize=True, sheet=0)
nt.assert_equal(self.sheet0, next(records))
records = io.read_xls(filepath, sanitize=True, sheet=1)
nt.assert_equal(self.sheet1, next(records))
kwargs = {'first_row': 1, 'first_col': 1}
records = io.read_xls(filepath, sanitize=True, sheet=2, **kwargs)
nt.assert_equal(self.sheet0, next(records))
records = io.read_xls(filepath, sanitize=True, sheet=3, **kwargs)
nt.assert_equal(self.sheet1, next(records))
def test_xls(self):
filepath = p.join(io.DATA_DIR, 'test.xlsx')
records = io.read_xls(filepath, sanitize=True, sheet=0)
nt.assert_equal(self.sheet0, next(records))
with open(filepath, 'r+b') as f:
records = io.read_xls(f, sanitize=True, sheet=0)
nt.assert_equal(self.sheet0, next(records))
records = io.read_xls(filepath, sanitize=True, sheet=1)
nt.assert_equal(self.sheet1, next(records))
kwargs = {'first_row': 1, 'first_col': 1}
records = io.read_xls(filepath, sanitize=True, sheet=2, **kwargs)
nt.assert_equal(self.sheet0, next(records))
records = io.read_xls(filepath, sanitize=True, sheet=3, **kwargs)
nt.assert_equal(self.sheet1, next(records))
def test_opened_files(self):
"""Test for reading open files"""
filepath = p.join(io.DATA_DIR, 'test.csv')
header = ['some_date', 'sparse_data', 'some_value', 'unicode_test']
with open(filepath, encoding='utf-8') as f:
records = io._read_csv(f, header) # pylint: disable=W0212
nt.assert_equal(self.sheet0_alt, next(records))
f = open(filepath, encoding='utf-8')
try:
records = io.read_csv(f, sanitize=True)
nt.assert_equal(self.sheet0_alt, next(records))
finally:
f.close()
f = open(filepath, 'rU', newline=None)
try:
records = io.read_csv(f, sanitize=True)
nt.assert_equal(self.sheet0_alt, next(records))
finally:
f.close()
filepath = p.join(io.DATA_DIR, 'test.xlsx')
with open(filepath, 'r+b') as f:
records = io.read_xls(f, sanitize=True, sheet=0)
nt.assert_equal(self.sheet0, next(records))
f = open(filepath, 'r+b')
try:
records = io.read_xls(f, sanitize=True, sheet=0)
nt.assert_equal(self.sheet0, next(records))
finally:
f.close()
def test_opened_files(self):
"""Test for reading open files"""
filepath = p.join(io.DATA_DIR, 'test.csv')
header = ['some_date', 'sparse_data', 'some_value', 'unicode_test']
with open(filepath, encoding='utf-8') as f:
records = io._read_csv(f, header) # pylint: disable=W0212
nt.assert_equal(self.sheet0_alt, next(records))
f = open(filepath, encoding='utf-8')
try:
records = io.read_csv(f, sanitize=True)
nt.assert_equal(self.sheet0_alt, next(records))
finally:
f.close()
f = open(filepath, 'rU', newline=None)
try:
records = io.read_csv(f, sanitize=True)
nt.assert_equal(self.sheet0_alt, next(records))
finally:
f.close()
filepath = p.join(io.DATA_DIR, 'test.xlsx')
with open(filepath, 'r+b') as f:
records = io.read_xls(f, sanitize=True, sheet=0)
nt.assert_equal(self.sheet0, next(records))
f = open(filepath, 'r+b')
try:
records = io.read_xls(f, sanitize=True, sheet=0)
nt.assert_equal(self.sheet0, next(records))
finally:
f.close()
def test_opened_files(self):
"""Test for reading open files"""
filepath = p.join(io.DATA_DIR, "test.csv")
with open(filepath, encoding="utf-8") as f:
records = io.read_csv(f, sanitize=True) # pylint: disable=W0212
nt.assert_equal(self.sheet0_alt, next(records))
f = open(filepath, encoding="utf-8")
try:
records = io.read_csv(f, sanitize=True)
nt.assert_equal(self.sheet0_alt, next(records))
finally:
f.close()
f = open(filepath, "rU", newline=None)
try:
records = io.read_csv(f, sanitize=True)
nt.assert_equal(self.sheet0_alt, next(records))
finally:
f.close()
filepath = p.join(io.DATA_DIR, "test.xlsx")
with open(filepath, "r+b") as f:
records = io.read_xls(f, sanitize=True, sheet=0)
nt.assert_equal(self.sheet0, next(records))
f = open(filepath, "r+b")
try:
records = io.read_xls(f, sanitize=True, sheet=0)
nt.assert_equal(self.sheet0, next(records))
finally:
f.close()
