def verify(self, t): v, X = fourier_transform.fourier_transform(t, self.f(t)) self.assertTrue(fourier_transform._is_constant(np.diff(v), positive=True)) assert_allclose(X, self.F(v), atol=self.atol) v, X = fourier_transform.fourier_transform(t, self.f(t), sign=-1) self.assertTrue(fourier_transform._is_constant(np.diff(v), positive=True)) assert_allclose(X, self.F(-v), atol=self.atol) v, X = fourier_transform.fourier_transform(t, self.f(t), rw_freq=5) self.assertTrue(fourier_transform._is_constant(np.diff(v), positive=True)) assert_allclose(X, self.F(v - 5), atol=self.atol)
def verify(self, t): v, X = fourier_transform.fourier_transform(t, self.f(t)) self.assertTrue( fourier_transform._is_constant(np.diff(v), positive=True)) assert_allclose(X, self.F(v), atol=self.atol) v, X = fourier_transform.fourier_transform(t, self.f(t), sign=-1) self.assertTrue( fourier_transform._is_constant(np.diff(v), positive=True)) assert_allclose(X, self.F(-v), atol=self.atol) v, X = fourier_transform.fourier_transform(t, self.f(t), rw_freq=5) self.assertTrue( fourier_transform._is_constant(np.diff(v), positive=True)) assert_allclose(X, self.F(v - 5), atol=self.atol)
def test_is_constant(): assert fourier_transform._is_constant([1, 1, 1]) assert fourier_transform._is_constant([-1, -1, -1]) assert not fourier_transform._is_constant([1, 1, 1, 0]) assert not fourier_transform._is_constant([1, 1, 1, 1], positive=False) assert fourier_transform._is_constant([1, 1, 1, 1], positive=True) assert fourier_transform._is_constant([-1, -1, -1], positive=False) assert fourier_transform._is_constant([-1, -1, -1], positive=False)