@@ -1782,7 +1782,6 @@ cdef class RandomState:
1782
1782
1783
1783
>>> s = np.random.complex_normal(size=1000)
1784
1784
"""
1785
- # TODO: What about loc??
1786
1785
cdef np .ndarray ogamma , orelation , oloc , randoms
1787
1786
cdef double * randoms_data
1788
1787
cdef double fgamma_r , fgamma_i , frelation_r , frelation_i , frho , f_v_real , f_v_imag , \
@@ -1823,7 +1822,7 @@ cdef class RandomState:
1823
1822
f_real *= sqrt (0.5 * f_v_real )
1824
1823
f_imag *= sqrt (0.5 * f_v_imag )
1825
1824
1826
- return PyComplex_FromDoubles (f_real , f_imag )
1825
+ return PyComplex_FromDoubles (floc_r + f_real , floc_i + f_imag )
1827
1826
1828
1827
randoms = < np .ndarray > np .empty (size , np .complex128 )
1829
1828
randoms_data = < double * > np .PyArray_DATA (randoms )
@@ -1837,8 +1836,8 @@ cdef class RandomState:
1837
1836
for i in range (n ):
1838
1837
random_gauss_zig_double_fill (& self .rng_state , 1 , & f_real )
1839
1838
random_gauss_zig_double_fill (& self .rng_state , 1 , & f_imag )
1840
- randoms_data [j + 1 ] = i_scale * (f_rho * f_real + i_r_scale * f_imag )
1841
- randoms_data [j ] = r_scale * f_real
1839
+ randoms_data [j + 1 ] = floc_i + i_scale * (f_rho * f_real + i_r_scale * f_imag )
1840
+ randoms_data [j ] = floc_r + r_scale * f_real
1842
1841
j += 2
1843
1842
1844
1843
return randoms
0 commit comments