diff --git a/unittests/test_table_importer.py b/unittests/test_table_importer.py index 49debb43f20704c301a77cf2386307c9d3de7d1b..746141771f0584c19dcdd962ed916b487a382d26 100644 --- a/unittests/test_table_importer.py +++ b/unittests/test_table_importer.py @@ -22,6 +22,7 @@ from tempfile import NamedTemporaryFile import numpy as np import pandas as pd +from caosadvancedtools.datainconsistency import DataInconsistencyError from caosadvancedtools.table_importer import (XLS_Importer, name_converter, yes_no_converter) @@ -70,6 +71,7 @@ class XLS_ImporterTest(unittest.TestCase): self.assertEqual(df_new.iloc[0].b, 5) def test_full(self): + """ test full run with example data """ tmp = NamedTemporaryFile(delete=False, suffix=".xlsx") tmp.close() self.valid_df.to_excel(tmp.name) @@ -81,3 +83,9 @@ class XLS_ImporterTest(unittest.TestCase): columns=['a', 'b', 'c', 'd']) df_new = self.importer.check_unique(df) self.assertEqual(df_new.shape[0], 1) + + def test_raise(self): + tmp = NamedTemporaryFile(delete=False, suffix=".lol") + tmp.close() + self.assertRaises(DataInconsistencyError, self.importer.read_xls, + tmp.name)