@@ -70,12 +70,12 @@ def test_get_spectra_complex():
70
70
r , _ , _ = utils .ar_generator (N = 2 ** 16 ) # It needs to be that long for
71
71
# the answers to converge
72
72
c , _ , _ = utils .ar_generator (N = 2 ** 16 )
73
- arsig1 = r + c * scipy . sqrt ( - 1 )
73
+ arsig1 = r + c * 1j
74
74
75
75
r , _ , _ = utils .ar_generator (N = 2 ** 16 )
76
76
c , _ , _ = utils .ar_generator (N = 2 ** 16 )
77
77
78
- arsig2 = r + c * scipy . sqrt ( - 1 )
78
+ arsig2 = r + c * 1j
79
79
avg_pwr1 .append ((arsig1 * arsig1 .conjugate ()).mean ())
80
80
avg_pwr2 .append ((arsig2 * arsig2 .conjugate ()).mean ())
81
81
@@ -118,7 +118,7 @@ def test_periodogram():
118
118
N = 1024
119
119
r , _ , _ = utils .ar_generator (N = N )
120
120
c , _ , _ = utils .ar_generator (N = N )
121
- arsig = r + c * scipy . sqrt ( - 1 )
121
+ arsig = r + c * 1j
122
122
123
123
f , c = tsa .periodogram (arsig )
124
124
npt .assert_equal (f .shape [0 ], N ) # Should be N, not the one-sided N/2 + 1
@@ -143,11 +143,11 @@ def test_periodogram_csd():
143
143
N = 1024
144
144
r , _ , _ = utils .ar_generator (N = N )
145
145
c , _ , _ = utils .ar_generator (N = N )
146
- arsig1 = r + c * scipy . sqrt ( - 1 )
146
+ arsig1 = r + c * 1j
147
147
148
148
r , _ , _ = utils .ar_generator (N = N )
149
149
c , _ , _ = utils .ar_generator (N = N )
150
- arsig2 = r + c * scipy . sqrt ( - 1 )
150
+ arsig2 = r + c * 1j
151
151
152
152
tseries = np .vstack ([arsig1 , arsig2 ])
153
153
0 commit comments