ByteParser now recognizes synthetic 'return None' blocks and treats them correctly.

2 files changed, 5 insertions, 5 deletions

diff --git a/test/test_arcs.py b/test/test_arcs.py
index 9b2c386f..47a69987 100644
--- a/test/test_arcs.py
+++ b/test/test_arcs.py
@@ -113,7 +113,7 @@ class SimpleArcTest(CoverageTest):
             arcz=".1 16 67 7.   .2 23 24 3. 45 5.", arcz_missing=""
             )
 
-    def XXX_dont_confuse_exit_and_else(self):
+    def test_dont_confuse_exit_and_else(self):
         self.check_coverage("""\
             def foo():
                 if foo:
@@ -121,7 +121,7 @@ class SimpleArcTest(CoverageTest):
                 else:
                     a = 5
                 return a
-            assert foo() == 3
+            assert foo() == 3 # 7
             """,
             arcz=".1 17 7.  .2 23 36 25 56 6.", arcz_missing="25 56"
             )
@@ -131,7 +131,7 @@ class SimpleArcTest(CoverageTest):
                     a = 3
                 else:
                     a = 5
-            foo()
+            foo() # 6
             """,
             arcz=".1 16 6.  .2 23 3. 25 5.", arcz_missing="25 5."
             )
diff --git a/test/test_parser.py b/test/test_parser.py
index 9f0f00b0..0dc7089c 100644
--- a/test/test_parser.py
+++ b/test/test_parser.py
@@ -67,7 +67,7 @@ class ParserTest(CoverageTest):
             1:0, 2:1, 3:1
             })
         
-    def XXX_missing_branch_to_excluded_code(self):
+    def test_missing_branch_to_excluded_code(self):
         cp = self.parse_source("""\
             if fooey:
                 a = 2
@@ -93,4 +93,4 @@ class ParserTest(CoverageTest):
                     a = 5
             b = 6
             """)
-        self.assertEqual(cp.exit_counts(), { 1:13, 2:1, 6:1 })
+        self.assertEqual(cp.exit_counts(), { 1:1, 2:1, 3:1, 6:1 })