diff options
Diffstat (limited to 'numpy/lib')
-rw-r--r-- | numpy/lib/function_base.py | 2 | ||||
-rw-r--r-- | numpy/lib/tests/test_function_base.py | 5 |
2 files changed, 6 insertions, 1 deletions
diff --git a/numpy/lib/function_base.py b/numpy/lib/function_base.py index 6f8df9b7b..dd5452723 100644 --- a/numpy/lib/function_base.py +++ b/numpy/lib/function_base.py @@ -38,7 +38,7 @@ def linspace(start, stop, num=50, endpoint=True, retstep=False): return array([]) if endpoint: if num == 1: - return array([start]) + return array([float(start)]) step = (stop-start)/float((num-1)) else: step = (stop-start)/float(num) diff --git a/numpy/lib/tests/test_function_base.py b/numpy/lib/tests/test_function_base.py index 3a603923a..372929447 100644 --- a/numpy/lib/tests/test_function_base.py +++ b/numpy/lib/tests/test_function_base.py @@ -87,6 +87,11 @@ class test_linspace(ScipyTestCase): y = list(linspace(0,1,2.5)) assert y == [0.0, 1.0] + def check_type(self): + t1 = linspace(0,1,0).dtype + t2 = linspace(0,1,1).dtype + assert_equal(t1, t2) + class test_amax(ScipyTestCase): def check_basic(self): a = [3,4,5,10,-3,-5,6.0] |