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