From ba93031da25edcd5175621b35f7c148c31218759 Mon Sep 17 00:00:00 2001
From: "Ilya.Kazakevich" <Ilya.Kazakevich@jetbrains.com>
Date: Mon, 13 Mar 2017 23:46:44 +0300
Subject: [PATCH] PY-7856: Fixed according to IDEA-CR-18091

---
 python/helpers/pycharm/docrunner.py | 6 +-----
 1 file changed, 1 insertion(+), 5 deletions(-)

diff --git a/python/helpers/pycharm/docrunner.py b/python/helpers/pycharm/docrunner.py
index 1a6c3cc64053..baa28f002c6c 100644
--- a/python/helpers/pycharm/docrunner.py
+++ b/python/helpers/pycharm/docrunner.py
@@ -180,11 +180,7 @@ class DocTestRunner(doctest.DocTestRunner):
         self.result.startTest(example)
         err = self._failure_header(test, example) +\
               self._checker.output_difference(example, got, self.optionflags)
-        expected = None
-        try:
-          expected = example.want
-        except KeyError:
-          pass
+        expected = getattr(example, "want", None)
         self.result.addFailure(example, err, expected=expected, actual=got)
 
       elif outcome is BOOM: