def test_eigen_circular_01(self): ce = Circular() mseed = 2963416 e_cue = ce.eigen_circular(5, ensemble='CUE', set_seed=True, seed=mseed) e_coe = ce.eigen_circular(10, ensemble='COE', set_seed=True, seed=mseed) e_cse = ce.eigen_circular(15, ensemble='CSE', set_seed=True, seed=mseed) n_cue = np.imag(e_cue).sum() n_coe = np.imag(e_coe).sum() n_cse = np.imag(e_cse).sum() self.assertTrue(n_cue+0.702464040915249 < self.epsilon) self.assertTrue(n_coe-0.023881520674723 < self.epsilon) self.assertTrue(np.abs(n_cse) < self.epsilon)
def test_eigen_circular_01(self): ce = Circular() mseed = 2963416 e_cue = ce.eigen_circular(5, ensemble='CUE', set_seed=True, seed=mseed) e_coe = ce.eigen_circular(10, ensemble='COE', set_seed=True, seed=mseed) e_cse = ce.eigen_circular(15, ensemble='CSE', set_seed=True, seed=mseed) n_cue = np.imag(e_cue).sum() n_coe = np.imag(e_coe).sum() n_cse = np.imag(e_cse).sum() self.assertTrue(n_cue - 1.198459165065781 < self.epsilon) self.assertTrue(n_coe + 0.068033777217730 < self.epsilon) self.assertTrue(np.abs(n_cse) < self.epsilon)
def test_spectral_density_01(self): N = 10 ce = Circular() mseed = 2963416 e_cue = ce.eigen_circular(N=N, ensemble='CUE', set_seed=True, seed=mseed) ergo = Ergodicity() sdensity = ergo.spectral_density(e_cue, ensemble_size=1, N=N) self.assertTrue(sdensity.shape == (31, 2)) self.assertTrue(sdensity[5:15, 1].sum() - 4.0 < self.epsilon) self.assertTrue( sdensity[15:27, 0].sum() - 12.700888156922527 < self.epsilon)