def test_exception(self, value, args, expected): with pytest.raises(expected): main = Main(value, args) main()
def test_normal(self, value, expected): main = Main(value) assert main() == expected