add additional tests for exception handling
This commit is contained in:
parent
46b31a9e8c
commit
54a677e0ce
|
|
@ -1,6 +1,7 @@
|
||||||
import os
|
import os
|
||||||
|
|
||||||
import pandas as pd
|
import pandas as pd
|
||||||
|
import pytest
|
||||||
|
|
||||||
from pandas_ods_reader import read_ods
|
from pandas_ods_reader import read_ods
|
||||||
|
|
||||||
|
|
@ -58,3 +59,16 @@ class TestOdsReader(object):
|
||||||
assert isinstance(df, pd.DataFrame)
|
assert isinstance(df, pd.DataFrame)
|
||||||
assert len(df) == 10
|
assert len(df) == 10
|
||||||
assert (len(df.columns) == 5)
|
assert (len(df.columns) == 5)
|
||||||
|
|
||||||
|
def test_wrong_id_type(self):
|
||||||
|
path = os.path.join(rsc, header_file)
|
||||||
|
with pytest.raises(ValueError) as e_info:
|
||||||
|
read_ods(path, 1.0)
|
||||||
|
assert e_info.match("Sheet id has to be either `str` or `int`")
|
||||||
|
|
||||||
|
def test_non_existent_sheet(self):
|
||||||
|
path = os.path.join(rsc, header_file)
|
||||||
|
sheet_name = "No_Sheet"
|
||||||
|
with pytest.raises(ValueError) as e_info:
|
||||||
|
read_ods(path, sheet_name)
|
||||||
|
assert e_info.match(f"There is no sheet named {sheet_name}")
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue