ENH: Validate type for list or tuple

Author: luxedo

scipy_doctest/impl.py

@@ -323,6 +323,12 @@ def check_output(self, want, got, optionflags):
             else:
                 return self.check_output(want_again, got_again, optionflags)
 
+        # Validate data type if list or tuple
+        is_list_or_tuple = (isinstance(a_want, (list, tuple)) or
+                            isinstance(a_got, (list, tuple)))
+        if is_list_or_tuple and type(a_want) is not type(a_got):
+            return False
+
         # ... and defer to numpy
         try:
             return self._do_check(a_want, a_got)

scipy_doctest/tests/failure_cases.py

@@ -1,4 +1,5 @@
-__all__ = ['func9', 'func10', 'iterable_length_1', 'iterable_length_2']
+__all__ = ['func9', 'func10', 'iterable_length_1', 'iterable_length_2',
+           'different_iteralbes_1', 'different_iteralbes_2']
 
 
 def func9():
@@ -30,3 +31,18 @@ def iterable_length_2():
     >>> [1, 2, 3]
     [1, 2]
     """
+
+
+def different_iteralbes_1():
+    """
+    >>> [0, 1, 2]
+    (0, 1, 2)
+    """
+
+
+def different_iteralbes_2():
+    """
+    >>> import numpy as np
+    >>> np.array([0, 1, 2])
+    [0, 1, 2]
+    """

scipy_doctest/tests/test_testmod.py

@@ -108,6 +108,15 @@ def test_wrong_lengths():
     assert res.failed == 2
 
 
+def test_different_iterables():
+    config = DTConfig()
+    res, _ = _testmod(failure_cases,
+                      strategy=[failure_cases.different_iteralbes_1,
+                                failure_cases.different_iteralbes_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