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)
