TST: add different lenghts iterables test

Author: luxedo

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