miklo
/
pandas_ods_reader
mirror of https://forge.citizen4.eu/miklo/pandas_ods_reader.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

additional checks for number of columns

This commit is contained in:
iuvbio 2019-06-08 14:01:10 +02:00
parent 9b4a272fd4
commit 46b31a9e8c
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
pandas_ods_reader/tests/test_read_ods.py
View File

@ -19,12 +19,14 @@ class TestOdsReader(object):
df = read_ods(path, 1)
assert isinstance(df, pd.DataFrame)
assert len(df) == 10
assert (len(df.columns) == 5)
def test_header_file_with_str(self):
path = os.path.join(rsc, header_file)
df = read_ods(path, "Sheet1")
assert isinstance(df, pd.DataFrame)
assert len(df) == 10
assert (len(df.columns) == 5)
def test_header_file_with_cols(self):
path = os.path.join(rsc, header_file)
@ -32,13 +34,15 @@ class TestOdsReader(object):
df = read_ods(path, "Sheet1", columns=columns)
assert list(df.columns) == columns
assert len(df) == 10
assert (len(df.columns) == 5)
def test_no_header_file_no_cols(self):
path = os.path.join(rsc, no_header_file)
df = read_ods(path, 1, headers=False)
assert list(df.columns) == [
"column_%s" % i for i in range(len(df.columns))]
f"column_{i}" for i in range(len(df.columns))]
assert len(df) == 10
assert (len(df.columns) == 5)
def test_no_header_file_with_cols(self):
path = os.path.join(rsc, no_header_file)
@ -46,9 +50,11 @@ class TestOdsReader(object):
df = read_ods(path, 1, headers=False, columns=columns)
assert list(df.columns) == columns
assert len(df) == 10
assert (len(df.columns) == 5)
def test_header_file_col_len(self):
path = os.path.join(rsc, col_len_file)
df = read_ods(path, 1)
assert isinstance(df, pd.DataFrame)
assert len(df) == 10
assert (len(df.columns) == 5)