def test_slice(self): """ Test slicing: f[0] should return the first component. """ s = Chebfun.from_function(segment) tools.assert_close(s[0], Chebfun.identity()) tools.assert_close(s[1], Chebfun(0.)) tools.assert_close(s[:], s)
def test_scalarvectormult(self): """ Possible to multiply scalar with vector chebfun. """ v = Chebfun.from_function(segment) s = np.sin(Chebfun.identity()) m = s * v tools.assert_close(m[0], s * v[0])
def test_scalarvectormult(self): """ Possible to multiply scalar with vector chebfun. """ v = Chebfun.from_function(segment) s = np.sin(Chebfun.identity()) m = s * v tools.assert_close(m[0], s*v[0])
def test_identity(self): c = Chebfun.identity() tools.assert_close(c, lambda x: x)
def test_restrict(self): x = Chebfun.identity() with self.assertRaises(ValueError): x.restrict([-2,0]) with self.assertRaises(ValueError): x.restrict([0,2])
def test_mismatch(self): c1 = Chebfun.identity() c2 = Chebfun.from_function(lambda x:x, domain=[2,3]) for op in [operator.add, operator.sub, operator.mul, operator.truediv]: with self.assertRaises(Chebfun.DomainMismatch): op(c1, c2)
def test_identity(self): c = Chebfun.identity() tools.assert_close(c, lambda x:x)
def test_restrict(self): x = Chebfun.identity() with self.assertRaises(ValueError): x.restrict([-2, 0]) with self.assertRaises(ValueError): x.restrict([0, 2])
def test_mismatch(self): c1 = Chebfun.identity() c2 = Chebfun.from_function(lambda x: x, domain=[2, 3]) for op in [operator.add, operator.sub, operator.mul, operator.truediv]: with self.assertRaises(Chebfun.DomainMismatch): op(c1, c2)