def test_fourier(self): seq1 = [0, 1] seq2 = [0, 1, 0] seq3 = [0, 1, 0, 1] r1 = [(0, 0.5, -0.0, 0.5, -0.0), (1, -0.5, 0.0, 0.5, numpy.pi)] r2 = [(0, 1/3, -0.0, 1/3, -0.0), (1, -1/6, 0.288675134594813, 0.3333333333333334, 2.0943951023931953)] r3 = [(0, 0.5, -0.0, 0.5, -0.0), (1, 0.0, -0.0, 0.0, -0.0), (2, -0.5, 0.0, 0.5, numpy.pi)] self.assertEqual(utils.fourier(seq1).to_records().tolist(), r1) self.assertEqual(utils.fourier(seq2).to_records().tolist(), r2) self.assertEqual(utils.fourier(seq3).to_records().tolist(), r3)
def fourier_analysis(self): return utils.fourier(self.to_tuple())