133
134
assert_equal(len(choices),3)
134
135
assert_equal(len(conditions),3)
136
class TestLogspace(TestCase):
137
def test_basic(self):
140
y = logspace(0,6,num=100)
141
assert(y[-1] == 10**6)
142
y = logspace(0,6,endpoint=0)
143
assert(y[-1] < 10**6)
144
y = logspace(0,6,num=7)
145
assert_array_equal(y,[1,10,100,1e3,1e4,1e5,1e6])
147
class TestLinspace(TestCase):
148
def test_basic(self):
151
y = linspace(2,10,num=100)
153
y = linspace(2,10,endpoint=0)
155
y,st = linspace(2,10,retstep=1)
156
assert_almost_equal(st,8/49.0)
157
assert_array_almost_equal(y,mgrid[2:10:50j],13)
159
def test_corner(self):
160
y = list(linspace(0,1,1))
162
y = list(linspace(0,1,2.5))
163
assert y == [0.0, 1.0]
166
t1 = linspace(0,1,0).dtype
167
t2 = linspace(0,1,1).dtype
168
t3 = linspace(0,1,2).dtype
172
137
class TestInsert(TestCase):
173
138
def test_basic(self):
754
719
[ 0.59541557, 0.87964135, 0.70543747],
755
720
[ 0.91084584, 0.84386844, 0.37068164]]))
722
def test_nanmin_allnan_on_axis(self):
723
assert_array_equal(isnan(nanmin([[nan]*2]*3, axis=1)),
758
727
class TestCorrCoef(TestCase):
759
728
def test_simple(self):
804
773
class TestKaiser(TestCase):
805
774
def test_simple(self):
806
assert_almost_equal(kaiser(0,1.0), array([]))
807
assert isnan(kaiser(1,1.0))
808
assert_almost_equal(kaiser(2,1.0), array([ 0.78984831, 0.78984831]))
809
assert_almost_equal(kaiser(5,1.0),
810
array([ 0.78984831, 0.94503323, 1. ,
811
0.94503323, 0.78984831]))
812
assert_almost_equal(kaiser(5,1.56789),
813
array([ 0.58285404, 0.88409679, 1. ,
814
0.88409679, 0.58285404]))
775
assert_almost_equal(kaiser(0, 1.0), array([]))
776
assert isfinite(kaiser(1, 1.0))
777
assert_almost_equal(kaiser(2, 1.0), array([ 0.78984831, 0.78984831]))
778
assert_almost_equal(kaiser(5, 1.0),
779
array([ 0.78984831, 0.94503323, 1. ,
780
0.94503323, 0.78984831]))
781
assert_almost_equal(kaiser(5, 1.56789),
782
array([ 0.58285404, 0.88409679, 1. ,
783
0.88409679, 0.58285404]))
816
785
def test_int_beta(self):