def figure1():
pl.figure(1, figsize=(6, 2))
pl.subplots_adjust(bottom=.25, top=.9, left=.1, right=.95, wspace=.5)
pl.clf()
steps = 100
X, Y = models.gp_puzzle_nub(steps=steps)
T = pl.arange(0., 1.0001, 1. / steps)
pl.subplot(1, 3, 1)
pl.text(-.35, 1, 'a)', va='bottom', ha='right')
pl.errorbar(data.puzzle_t,
data.puzzle_x,
pl.sqrt(data.puzzle_V) * 1.96,
marker='o',
color='black',
linestyle='dotted')
graphics.my_plot(T, X, color='red', alpha=.5, linewidth=2)
pl.axis([-.1, 1.1, -.1, 1.1])
pl.xticks([0, 1])
pl.yticks([0, 1])
pl.xlabel('$t$')
pl.ylabel('$x(t)$')
pl.subplot(1, 3, 2)
pl.text(-.35, 1, 'b)', va='bottom', ha='right')
pl.errorbar(data.puzzle_t,
data.puzzle_y,
pl.sqrt(data.puzzle_V) * 1.96,
marker='o',
color='black',
linestyle='dotted')
graphics.my_plot(T, Y, color='red', alpha=.5, linewidth=2)
pl.axis([-.1, 1.1, -.1, 1.1])
pl.xticks([0, 1])
pl.yticks([0, 1])
pl.xlabel('$t$')
pl.ylabel('$y(t)$')
pl.subplot(1, 3, 3)
pl.text(-.35, 1, 'c)', va='bottom', ha='right')
pl.plot(data.puzzle_x,
data.puzzle_y,
marker='o',
color='black',
linestyle='dotted')
graphics.my_plot(X, Y, color='red', alpha=.5, linewidth=2)
pl.axis([-.1, 1.1, -.1, 1.1])
pl.xticks([0, 1])
pl.yticks([0, 1])
pl.xlabel('$x(t)$')
pl.ylabel('$y(t)$')
pl.savefig('../tex/fig1.pdf')
pl.savefig('../fig1.png')
def plot_nub(a, b, diff_degree=2., amp=1., scale=1.5, steps=100, **addl_plot_params):
""" Plot a puzzle nub from a to b"""
X, Y = models.gp_puzzle_nub(diff_degree, amp, scale, steps)
xy = pl.array([X, Y])
a = pl.array(a)
b = pl.array(b)
r = pl.norm(b-a)
cosT, sinT = (b-a)/r
R = pl.array([[cosT, -sinT],
[sinT, cosT]])
xy = a + r*pl.dot(R, xy).T
X = xy[:, 0]
Y = xy[:, 1]
my_plot(X, Y, **addl_plot_params)
def figure1():
pl.figure(1, figsize=(6, 2))
pl.subplots_adjust(bottom=.25, top=.9, left=.1, right=.95, wspace=.5)
pl.clf()
steps = 100
X, Y = models.gp_puzzle_nub(steps=steps)
T = pl.arange(0., 1.0001, 1. / steps)
pl.subplot(1,3,1)
pl.text(-.35, 1, 'a)', va='bottom', ha='right')
pl.errorbar(data.puzzle_t, data.puzzle_x, pl.sqrt(data.puzzle_V)*1.96,
marker='o', color='black', linestyle='dotted')
graphics.my_plot(T, X, color='red', alpha=.5, linewidth=2)
pl.axis([-.1, 1.1, -.1, 1.1])
pl.xticks([0,1])
pl.yticks([0,1])
pl.xlabel('$t$')
pl.ylabel('$x(t)$')
pl.subplot(1,3,2)
pl.text(-.35, 1, 'b)', va='bottom', ha='right')
pl.errorbar(data.puzzle_t, data.puzzle_y, pl.sqrt(data.puzzle_V)*1.96,
marker='o', color='black', linestyle='dotted')
graphics.my_plot(T, Y, color='red', alpha=.5, linewidth=2)
pl.axis([-.1, 1.1, -.1, 1.1])
pl.xticks([0,1])
pl.yticks([0,1])
pl.xlabel('$t$')
pl.ylabel('$y(t)$')
pl.subplot(1,3,3)
pl.text(-.35, 1, 'c)', va='bottom', ha='right')
pl.plot(data.puzzle_x, data.puzzle_y,
marker='o', color='black', linestyle='dotted')
graphics.my_plot(X, Y, color='red', alpha=.5, linewidth=2)
pl.axis([-.1, 1.1, -.1, 1.1])
pl.xticks([0,1])
pl.yticks([0,1])
pl.xlabel('$x(t)$')
pl.ylabel('$y(t)$')
pl.savefig('../tex/fig1.pdf')
pl.savefig('../fig1.png')
def test_models(self):
X, Y = models.gp_puzzle_nub()
assert -.01 <= X[0] <= .01
assert -.01 <= Y[0] <= .01
def test_models(self):
X, Y = models.gp_puzzle_nub()
assert -.01 <= X[0] <= .01
assert -.01 <= Y[0] <= .01
