1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
|
from numpy.testing import *
from numpy import logspace, linspace
class TestLogspace(TestCase):
def test_basic(self):
y = logspace(0,6)
assert(len(y)==50)
y = logspace(0,6,num=100)
assert(y[-1] == 10**6)
y = logspace(0,6,endpoint=0)
assert(y[-1] < 10**6)
y = logspace(0,6,num=7)
assert_array_equal(y,[1,10,100,1e3,1e4,1e5,1e6])
class TestLinspace(TestCase):
def test_basic(self):
y = linspace(0,10)
assert(len(y)==50)
y = linspace(2,10,num=100)
assert(y[-1] == 10)
y = linspace(2,10,endpoint=0)
assert(y[-1] < 10)
def test_corner(self):
y = list(linspace(0,1,1))
assert y == [0.0], y
y = list(linspace(0,1,2.5))
assert y == [0.0, 1.0]
def test_type(self):
t1 = linspace(0,1,0).dtype
t2 = linspace(0,1,1).dtype
t3 = linspace(0,1,2).dtype
assert_equal(t1, t2)
assert_equal(t2, t3)
|
