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

if sheet does not exist, raise KeyError instead of ValueError

This commit is contained in:
iuvbio 2020-02-23 17:24:08 +01:00
parent 35724dde06
commit e1c602d01a
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
pandas_ods_reader/parser.py
View File

@ -14,7 +14,7 @@ def load_ods(doc, sheet_id, headers=True, columns=None):
if isinstance(sheet_id, str): if isinstance(sheet_id, str):
sheets = [sheet.name for sheet in doc.sheets] sheets = [sheet.name for sheet in doc.sheets]
if sheet_id not in sheets: if sheet_id not in sheets:
raise ValueError("There is no sheet named {}".format(sheet_id)) raise KeyError("There is no sheet named {}".format(sheet_id))
sheet_id = sheets.index(sheet_id) + 1 sheet_id = sheets.index(sheet_id) + 1
sheet = doc.sheets[sheet_id - 1] sheet = doc.sheets[sheet_id - 1]
df_dict = OrderedDict() df_dict = OrderedDict()

2
tests/test_read_ods.py
View File

@ -107,7 +107,7 @@ class TestOdsReader:
path = rsc / header_file path = rsc / header_file
sheet_name = "No_Sheet" sheet_name = "No_Sheet"
with pytest.raises(ValueError) as e_info: with pytest.raises(KeyError) as e_info:
read_ods(path, sheet_name) read_ods(path, sheet_name)
assert e_info.match(f"There is no sheet named {sheet_name}") assert e_info.match(f"There is no sheet named {sheet_name}")