TST: Fix numpy#6542: Add test for non-iterable input...

gkBCCN · gkBCCN · commit 8e05d78c7214 · 2016-02-25T12:34:41.000+01:00
...to hstack, vstack, and stack
diff --git a/numpy/core/tests/test_shape_base.py b/numpy/core/tests/test_shape_base.py
@@ -120,6 +120,9 @@ def test_3D_array(self):
 
 
 class TestHstack(TestCase):
+    def test_non_iterable(self):
+        assert_raises(TypeError, hstack, 1)
+
     def test_0D_array(self):
         a = array(1)
         b = array(2)
@@ -143,6 +146,9 @@ def test_2D_array(self):
 
 
 class TestVstack(TestCase):
+    def test_non_iterable(self):
+        assert_raises(TypeError, vstack, 1)
+
     def test_0D_array(self):
         a = array(1)
         b = array(2)
@@ -265,6 +271,9 @@ def test_concatenate(self):
 
 
 def test_stack():
+    # non-iterable input
+    assert_raises(TypeError, stack, 1)
+
     # 0d input
     for input_ in [(1, 2, 3),
                    [np.int32(1), np.int32(2), np.int32(3)],
