def test_plot_complex(self):
c = np.exp(1j*Chebfun.identity(domain=[-np.pi,np.pi]))
xs,ys,xi,yi,d = plot_data(c, plot_res)
self.assertEqual(d, 2, "dimension is two for complex chebfun")
for X,Y in [(xs,ys), (xi,yi)]:
dist = np.square(X) + np.square(Y)
npt.assert_allclose(dist, 1, err_msg="The plot should be a circle")
plot(c)
def test_plot_complex(self):
c = np.exp(1j * Chebfun.identity(domain=[-np.pi, np.pi]))
xs, ys, xi, yi, d = plot_data(c, plot_res)
self.assertEqual(d, 2, "dimension is two for complex chebfun")
for X, Y in [(xs, ys), (xi, yi)]:
dist = np.square(X) + np.square(Y)
npt.assert_allclose(dist, 1, err_msg="The plot should be a circle")
plot(c)
def test_ax_arg(self):
plt.clf()
lw = 2
ff = self.p
gg = self.p + 1
hh = self.p - 1
ax = plot(ff, color="k", linestyle="-", linewidth=lw)
plot(gg, ax=ax, color="r", linestyle="--", linewidth=lw)
plot(hh, ax=ax, color="r", linestyle="--", linewidth=lw)
def test_plot_interpolation_points(self):
plt.clf()
plot(self.p, with_interpolation_points=True)
a = plt.gca()
self.assertEqual(len(a.lines),2)
plt.clf()
plot(self.p, with_interpolation_points=False)
a = plt.gca()
self.assertEqual(len(a.lines),1)
def test_ax_arg(self):
plt.clf()
lw = 2
ff = self.p
gg = self.p + 1
hh = self.p - 1
ax = plot(ff, color="k", linestyle="-", linewidth=lw)
plot(gg, ax=ax, color="r", linestyle="--", linewidth=lw)
plot(hh, ax=ax, color="r", linestyle="--", linewidth=lw)
def test_plot_interpolation_points(self):
plt.clf()
plot(self.p, with_interpolation_points=True)
a = plt.gca()
self.assertEqual(len(a.lines), 2)
plt.clf()
plot(self.p, with_interpolation_points=False)
a = plt.gca()
self.assertEqual(len(a.lines), 1)
def test_plot_circle(self):
T = .5
def cirper(x):
return tools.circle(x, period=T)
c = Chebfun.from_function(cirper, domain=[0,T])
xs,ys,xi,yi,d = plot_data(c, plot_res)
self.assertEqual(d, 2,)
for X,Y in [(xs,ys), (xi,yi)]:
dist = np.square(X) + np.square(Y)
npt.assert_allclose(dist, 1, err_msg="The plot should be a circle")
plot(c)
def test_plot_circle(self):
T = .5
def cirper(x):
return tools.circle(x, period=T)
c = Chebfun.from_function(cirper, domain=[0, T])
xs, ys, xi, yi, d = plot_data(c, plot_res)
self.assertEqual(
d,
2,
)
for X, Y in [(xs, ys), (xi, yi)]:
dist = np.square(X) + np.square(Y)
npt.assert_allclose(dist, 1, err_msg="The plot should be a circle")
plot(c)
def test_too_many_dimensions(self):
c = Chebfun.from_data(np.random.random_sample([3,4]))
with self.assertRaises(ValueError):
plot(c)
def test_plot(self):
xs,ys,xi,yi,d = plot_data(self.p, plot_res)
self.assertEqual(d, 1)
npt.assert_allclose(ys, tools.f(xs))
plot(self.p)
def test_too_many_dimensions(self):
c = Chebfun.from_data(np.random.random_sample([3, 4]))
with self.assertRaises(ValueError):
plot(c)
def test_plot(self):
xs, ys, xi, yi, d = plot_data(self.p, plot_res)
self.assertEqual(d, 1)
npt.assert_allclose(ys, tools.f(xs))
plot(self.p)
