test corner cases

life4 · Apr 21, 2020 · acd8d1c · acd8d1c
1 parent 1b4c703
commit acd8d1c
Showing 1 changed file with 19 additions and 0 deletions.
diff --git a/tests/test_linter/test_extractors/test_exceptions.py b/tests/test_linter/test_extractors/test_exceptions.py
@@ -125,3 +125,22 @@ def f():
     print(ast.dump(tree))
     func_tree = tree.body[-1].body
     tuple(get_exceptions(body=func_tree))
+
+
+def test_inference_subcontracts():
+    text = """
+        @deal.raises(SomeError)     # actual contract
+        @deal.raises(1)             # ignore junk
+        @deal.post(lambda _: 1)     # ignore other contracts
+        def subf():
+            return 1
+
+        @deal.raises(KeyError)
+        def f():
+            b = subf()
+    """
+    tree = astroid.parse(dedent(text))
+    print(tree.repr_tree())
+    func_tree = tree.body[-1].body
+    returns = tuple(r.value for r in get_exceptions(body=func_tree))
+    assert returns == ('SomeError', )