From baf9af523a5b6a7121e88bf6580683d688b25dd3 Mon Sep 17 00:00:00 2001 From: Piotr Maslanka <piotr.maslanka@henrietta.com.pl> Date: Sat, 17 Feb 2018 00:11:49 +0100 Subject: [PATCH] fixes #14 --- tests/test_coding/test_rethrow.py | 12 +++++++++++- 1 file changed, 11 insertions(+), 1 deletion(-) diff --git a/tests/test_coding/test_rethrow.py b/tests/test_coding/test_rethrow.py index eb6f1694..46ee7af4 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)) -- GitLab