diff --git a/tests/test_coding/test_rethrow.py b/tests/test_coding/test_rethrow.py index eb6f169402a6cda8ea08a30e84e1999a45f88062..46ee7af4cd6b74a0b202b25a42928eee9c556e5a 100644 --- a/tests/test_coding/test_rethrow.py +++ b/tests/test_coding/test_rethrow.py @@ -54,7 +54,7 @@ class TestStuff(unittest.TestCase): self.assertRaises(WTFException1, lambda: provide(NameError)) self.assertRaises(WTFException2, lambda: provide(TypeError)) - def test_issue_14(self): + def test_issue_14a(self): @rethrow_as(((NameError, ValueError), TypeError)) def ro(p): @@ -63,3 +63,13 @@ class TestStuff(unittest.TestCase): self.assertRaises(TypeError, lambda: ro(NameError)) self.assertRaises(TypeError, lambda: ro(ValueError)) self.assertRaises(RuntimeError, lambda: ro(RuntimeError)) + + def test_issue_14b(self): + + @rethrow_as((((NameError, ValueError), TypeError), )) + def ro(p): + raise p() + + self.assertRaises(TypeError, lambda: ro(NameError)) + self.assertRaises(TypeError, lambda: ro(ValueError)) + self.assertRaises(RuntimeError, lambda: ro(RuntimeError))