def test_chebpolyfit(self): N = 32 data = np.random.rand(N - 1, 2) coeffs = Chebfun.polyfit(data) result = Chebfun.polyval(coeffs) npt.assert_allclose(data, result)
def test_list(self): data = [[1., 2], [3, 4]] adata = np.array(data) result = Chebfun.polyfit(data) expected = Chebfun.polyfit(adata) npt.assert_array_almost_equal(result, expected)
def test_chebpolyfitval(self, N=64): data = np.random.rand(N - 1, 2) computed = Chebfun.polyval(Chebfun.polyfit(data)) npt.assert_allclose(computed, data)
def test_scalar(self): for datalen in [1, 3]: coeffs = Chebfun.polyfit(np.ones([datalen])) self.assertEqual(len(coeffs.shape), 1)
def test_vector(self): for datalen in [1, 3]: coeffs = Chebfun.polyfit(np.ones([datalen, 2])) self.assertEqual(len(coeffs.shape), 2)
def test_chebpolyfit(self): N = 32 data = np.random.rand(N-1, 2) coeffs = Chebfun.polyfit(data) result = Chebfun.polyval(coeffs) npt.assert_allclose(data, result)
def test_chebpolyfitval(self, N=64): data = np.random.rand(N-1, 2) computed = Chebfun.polyval(Chebfun.polyfit(data)) npt.assert_allclose(computed, data)
def test_list(self): data = [[1.,2], [3,4]] adata = np.array(data) result = Chebfun.polyfit(data) expected = Chebfun.polyfit(adata) npt.assert_array_almost_equal(result, expected)
def test_vector(self): for datalen in [1,3]: coeffs = Chebfun.polyfit(np.ones([datalen, 2])) self.assertEqual(len(coeffs.shape), 2)
def test_scalar(self): for datalen in [1,3]: coeffs = Chebfun.polyfit(np.ones([datalen])) self.assertEqual(len(coeffs.shape), 1)