summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--Lib/doctest.py10
-rw-r--r--Lib/test/test_doctest.py19
2 files changed, 29 insertions, 0 deletions
diff --git a/Lib/doctest.py b/Lib/doctest.py
index 5d315cc..aba98dc 100644
--- a/Lib/doctest.py
+++ b/Lib/doctest.py
@@ -454,6 +454,9 @@ class Example:
def __ne__(self, other):
return not self == other
+ def __hash__(self):
+ return hash((self.source, self.want, self.lineno, self.indent,
+ self.exc_msg))
class DocTest:
"""
@@ -517,6 +520,9 @@ class DocTest:
def __ne__(self, other):
return not self == other
+ def __hash__(self):
+ return hash((self.docstring, self.name, self.filename, self.lineno))
+
# This lets us sort tests by name:
def __lt__(self, other):
if not isinstance(other, DocTest):
@@ -2247,6 +2253,10 @@ class DocTestCase(unittest.TestCase):
def __ne__(self, other):
return not self == other
+ def __hash__(self):
+ return hash((self._dt_optionflags, self._dt_setUp, self._dt_tearDown,
+ self._dt_checker))
+
def __repr__(self):
name = self._dt_test.name.split('.')
return "%s (%s)" % (name[-1], '.'.join(name[:-1]))
diff --git a/Lib/test/test_doctest.py b/Lib/test/test_doctest.py
index ac79e02..cdcd389 100644
--- a/Lib/test/test_doctest.py
+++ b/Lib/test/test_doctest.py
@@ -259,6 +259,21 @@ unless it's `None`:
>>> e = doctest.Example('raise X()', '', exc_msg)
>>> e.exc_msg
'\n'
+
+Compare `Example`:
+ >>> example = doctest.Example('print 1', '1\n')
+ >>> same_example = doctest.Example('print 1', '1\n')
+ >>> other_example = doctest.Example('print 42', '42\n')
+ >>> example == same_example
+ True
+ >>> example != same_example
+ False
+ >>> hash(example) == hash(same_example)
+ True
+ >>> example == other_example
+ False
+ >>> example != other_example
+ True
"""
def test_DocTest(): r"""
@@ -362,6 +377,8 @@ Compare `DocTest`:
True
>>> test != same_test
False
+ >>> hash(test) == hash(same_test)
+ True
>>> docstring = '''
... >>> print 42
... 42
@@ -383,6 +400,8 @@ Compare `DocTestCase`:
True
>>> test_case != same_test_case
False
+ >>> hash(test_case) == hash(same_test_case)
+ True
>>> test == other_test_case
False
>>> test != other_test_case