diff --git a/tests/test_vshell.py b/tests/test_vshell.py index 7aef83f..3bdbe92 100644 --- a/tests/test_vshell.py +++ b/tests/test_vshell.py @@ -130,7 +130,7 @@ def test_log(self): def test_pferror_class(self): """Tests the PFError exception class.""" with self.assertRaises(pfsense_vshell.PFError): - raise pfsense_vshell.PFError + raise pfsense_vshell.PFError(1, "Test error message") if __name__ == '__main__':