Merge pull request #162 from luxedo/FIX/161

FIX: iterable checks longest

Author: ev-br

scipy_doctest/impl.py

@@ -2,6 +2,7 @@
 import warnings
 import doctest
 from doctest import NORMALIZE_WHITESPACE, ELLIPSIS, IGNORE_EXCEPTION_DETAIL
+from itertools import zip_longest
 
 import numpy as np
 
@@ -328,7 +329,7 @@ def check_output(self, want, got, optionflags):
         except Exception:
             # heterog tuple, eg (1, np.array([1., 2.]))
             try:
-                return all(self._do_check(w, g) for w, g in zip(a_want, a_got))
+                return all(self._do_check(w, g) for w, g in zip_longest(a_want, a_got))
             except (TypeError, ValueError):
                 return False
 

scipy_doctest/tests/failure_cases.py

@@ -1,4 +1,5 @@
-__all__ = ['func9', 'func10']
+__all__ = ['func9', 'func10', 'iterable_length_1', 'iterable_length_2']
+
 
 def func9():
     """
@@ -15,3 +16,17 @@ def func10():
     >>> import numpy as np
     >>> np.arraY([1, 2, 3])
     """
+
+
+def iterable_length_1():
+    """
+    >>> [1, 2, 3]
+    [1, 2, 3, 4]
+    """
+
+
+def iterable_length_2():
+    """
+    >>> [1, 2, 3]
+    [1, 2]
+    """

scipy_doctest/tests/test_testmod.py

@@ -99,6 +99,15 @@ def test_user_context():
                 config=config)
 
 
+def test_wrong_lengths():
+    config = DTConfig()
+    res, _ = _testmod(failure_cases,
+                      strategy=[failure_cases.iterable_length_1,
+                                failure_cases.iterable_length_2],
+                      config=config)
+    assert res.failed == 2
+
+
 class TestLocalFiles:
     def test_local_files(self):
         # A doctest tries to open a local file. Test that it works