From d4261cafa36cb9b531e939e3334ac8dab575d5a8 Mon Sep 17 00:00:00 2001
From: coolreader18 <33094578+coolreader18@users.noreply.github.com>
Date: Tue, 17 Dec 2019 22:44:31 -0600
Subject: [PATCH] Add new test for cause

---
 tests/snippets/try_exceptions.py | 18 +++++++++++++++++-
 1 file changed, 17 insertions(+), 1 deletion(-)

diff --git a/tests/snippets/try_exceptions.py b/tests/snippets/try_exceptions.py
index c646e09370..1f46caae3e 100644
--- a/tests/snippets/try_exceptions.py
+++ b/tests/snippets/try_exceptions.py
@@ -250,7 +250,23 @@ try:
     except ZeroDivisionError as ex:
         raise NameError from ex
 except NameError as ex2:
-    pass
+    assert isinstance(ex2.__cause__, ZeroDivisionError)
+else:
+    assert False, "no raise"
+
+
+try:
+    try:
+        try:
+            raise ZeroDivisionError
+        except ZeroDivisionError as ex:
+            raise NameError from ex
+    except NameError:
+        raise
+except NameError as ex2:
+    assert isinstance(ex2.__cause__, ZeroDivisionError)
+else:
+    assert False, "no raise"
 
 
 # the else clause requires at least one except clause: