Tests of statements after raise and return

Author: nedbat

tests/test_coverage.py

@@ -368,6 +368,16 @@ def test_raise(self):
             """,
             [1,2,5,6], "")
 
+    def test_raise_followed_by_statement(self):
+        self.check_coverage("""\
+            try:
+                raise Exception("hello")
+                a = 3
+            except:
+                pass
+            """,
+            [1,2,3,4,5], "3")
+
     def test_return(self):
         self.check_coverage("""\
             def fn():
@@ -401,6 +411,18 @@ def fn():
             """,
             [1,2,3,7,8], "")
 
+    def test_return_followed_by_statement(self):
+        self.check_coverage("""\
+            def fn():
+                a = 1
+                return a
+                a = 2
+
+            x = fn()
+            assert(x == 1)
+            """,
+            [1,2,3,4,6,7], "4")
+
     def test_yield(self):
         self.check_coverage("""\
             def gen():