diff options
Diffstat (limited to 'numpy/lib/tests/test_function_base.py')
| -rw-r--r-- | numpy/lib/tests/test_function_base.py | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/numpy/lib/tests/test_function_base.py b/numpy/lib/tests/test_function_base.py index b1d5ec450..c3eaa9dd2 100644 --- a/numpy/lib/tests/test_function_base.py +++ b/numpy/lib/tests/test_function_base.py @@ -93,6 +93,13 @@ class test_linspace(NumpyTestCase): t3 = linspace(0,1,2).dtype assert_equal(t1, t2) assert_equal(t2, t3) + +class test_insert(NumpyTestCase): + def check_basic(self): + a = [1,2,3] + assert_equal(insert(a,0,1), [1,1,2,3]) + assert_equal(insert(a,3,1), [1,2,3,1]) + assert_equal(insert(a,[1,1,1],[1,2,3]), [1,1,2,3,2,3]) class test_amax(NumpyTestCase): def check_basic(self): |