def test_index_out_of_length(self):
s = ''
i = 1
self.assertEqual(
bracket.bracket(s, i),
'Passed index is out of string length',
'Asserting that index is on range')
def test_iter(self):
composers = ["Ludwig", "Copland", "Bernstein", "Britten", "Schubert", "Chopin"]
b1 = rankedElement(composers[0])
b2 = rankedElement(composers[1])
b3 = rankedElement(composers[2])
b4 = rankedElement(composers[3])
b5 = rankedElement(composers[4])
b6 = rankedElement(composers[5])
b7 = branchedElement(b1, b2)
b8 = branchedElement(b3, b4)
b9 = branchedElement(b5, b7)
b10 = branchedElement(b6, b8)
top = branchedElement(b9, b10)
br = bracket.bracket(top)
elements = [b1, b2, b3, b4, b5, b6, b7, b8, b9, b10, top]
for e in br:
if e in elements:
elements.remove(e)
else:
raise ValueError("Found element " + str(e) + " in this tree for some reason!")
self.assertEqual(len(elements), 0)
def test_expected_user_outputs(self):
s = '[ABC[23]][89]'
i = 4
self.assertEqual(
bracket.bracket(s, i),
'The opening bracket at index: 4, closes at index: 7',
'Asserting that the correct message is displayed'
)
xhigh = -6 + xmin
xplot.set_xlim(xlow, xhigh)
zplot.set_xlim(x_to_z(xlow), x_to_z(xhigh))
xticks = [6, 4, 2]
zticks = [1, 3, 5, 7]
plotticks = xticks + [z_to_x(z) for z in zticks]
xplot.set_xticks(xticks)
zplot.set_xticks(zticks)
xplot.set_xticks(plotticks)
xplot.set_xticklabels(['$%i$' % tick for tick in xticks] +
['$%i$' % tick for tick in zticks])
zplot.set_xticks([])
bracket.bracket(xplot, -.01, (xlow - 2) / (xlow - xhigh) + 0.007, -.06, .06,
r'$x/R$')
bracket.bracket(xplot, (xlow - xmin) / (xlow - xhigh) - 0.013, 1.01, -.06, .06,
r'$z/R$')
twod_plot.set_xlim(zmin, zmax)
twod_plot.set_ylim(-rmax, rmax)
#fig.subplots_adjust(hspace=0.001)
twod_plot.set_aspect('equal')
g3mc = read_triplet(ff, 'mc')[:, int(center / dx):-1]
rpoints = len(g3mc[:, 0])
zpoints = len(g3mc[0, :])
r = arange(0, rpoints * dx, dx)
z = arange(center, center + zpoints * dx, dx)
Z, R = meshgrid(z, r)
def x_to_z(x):
return 2 - x
zmax_lineplot = 6.
xmax_lineplot = 4.
xplot.set_xlim(zmax_lineplot, -xmax_lineplot)
xplot.set_xticks([6, 4, 2, 0, -2, -4])
xplot.set_xticklabels(["$6$", "$4$", "$2~~0$", "$2$", "$4$", "$6$"])
zplot.set_xlim(x_to_z(xplot.get_xlim()[0]), x_to_z(xplot.get_xlim()[1]))
zplot.set_xticks([])
#xplot.set_ylim(0)
bracket.bracket(xplot, -.01, xmax_lineplot / (xmax_lineplot + zmax_lineplot),
-.06, .06, r'$x/R$')
bracket.bracket(zplot, xmax_lineplot / (xmax_lineplot + zmax_lineplot), 1.01,
-.06, .06, r'$z/R$')
twod_plot.set_xlim(-1, 6)
twod_plot.set_ylim(-ymax, ymax)
sphere = Circle((0, 0), 1, color='slategray')
twod_plot.add_artist(sphere)
#fig.subplots_adjust(hspace=0.001)
twod_plot.set_aspect('equal')
g2mc = read_walls_mc(ff)
rbins = int(round(rmax / dx))
zposbins = int(round(zmax / dx))
znegbins = int(round(-zmin / dx))
def z_to_x(z):
return 2 - z
def x_to_z(x):
return 2 - x
zmax_lineplot = 6.
xmax_lineplot = 4.
xplot.set_xlim(zmax_lineplot, -xmax_lineplot)
xplot.set_xticks([6, 4, 2, 0, -2, -4])
xplot.set_xticklabels(["$6$", "$4$", "$2~~0$", "$2$", "$4$", "$6$"])
zplot.set_xlim(x_to_z(xplot.get_xlim()[0]), x_to_z(xplot.get_xlim()[1]))
zplot.set_xticks([])
#xplot.set_ylim(0)
bracket.bracket(xplot, -.01, xmax_lineplot/(xmax_lineplot+zmax_lineplot), -.06, .06, r'$x/R$')
bracket.bracket(zplot, xmax_lineplot/(xmax_lineplot+zmax_lineplot), 1.01, -.06, .06, r'$z/R$')
twod_plot.set_xlim(-1, 6)
twod_plot.set_ylim(-ymax, ymax)
sphere = Circle((0, 0), 1, color='slategray')
twod_plot.add_artist(sphere)
#fig.subplots_adjust(hspace=0.001)
twod_plot.set_aspect('equal')
g2mc = read_walls_mc(ff)
rbins = round(rmax/dx)
zposbins = round(zmax/dx)
znegbins = round(-zmin/dx)
def x_to_z(x):
return 2 - x
zmax_lineplot = 6.0
xmax_lineplot = 4.0
xplot.set_xlim(zmax_lineplot, -xmax_lineplot)
xplot.set_xticks([6, 4, 2, 0, -2, -4])
xplot.set_xticklabels(["$6$", "$4$", "$2~~0$", "$2$", "$4$", "$6$"])
zplot.set_xlim(x_to_z(xplot.get_xlim()[0]), x_to_z(xplot.get_xlim()[1]))
zplot.set_xticks([])
# xplot.set_ylim(0)
bracket.bracket(xplot, -0.01, xmax_lineplot / (xmax_lineplot + zmax_lineplot), -0.06, 0.06, r"$x/R$")
bracket.bracket(zplot, xmax_lineplot / (xmax_lineplot + zmax_lineplot), 1.01, -0.06, 0.06, r"$z/R$")
twod_plot.set_xlim(-1, 6)
twod_plot.set_ylim(-ymax, ymax)
sphere = Circle((0, 0), 1, color="slategray")
twod_plot.add_artist(sphere)
# fig.subplots_adjust(hspace=0.001)
twod_plot.set_aspect("equal")
g2mc = read_walls_mc(ff)
rbins = round(rmax / dx)
zposbins = round(zmax / dx)
znegbins = round(-zmin / dx)
xhigh = -6+xmin
xplot.set_xlim(xlow, xhigh)
zplot.set_xlim(x_to_z(xlow), x_to_z(xhigh))
xticks = [6, 4, 2]
zticks = [1, 3, 5, 7]
plotticks = xticks + [z_to_x(z) for z in zticks]
xplot.set_xticks(xticks)
zplot.set_xticks(zticks)
xplot.set_xticks(plotticks)
xplot.set_xticklabels(['$%i$' %tick for tick in xticks] +
['$%i$' %tick for tick in zticks])
zplot.set_xticks([])
bracket.bracket(xplot, -.01, (xlow - 2)/(xlow - xhigh) + 0.007, -.06, .06, r'$x/R$')
bracket.bracket(xplot, (xlow - xmin)/(xlow - xhigh) - 0.013, 1.01, -.06, .06, r'$z/R$')
twod_plot.set_xlim(zmin, zmax)
twod_plot.set_ylim(-rmax, rmax)
#fig.subplots_adjust(hspace=0.001)
twod_plot.set_aspect('equal')
g3mc = read_triplet(ff, 'mc')[:, center/dx:-1]
rpoints = len(g3mc[:,0])
zpoints = len(g3mc[0,:])
r = arange(0, rpoints*dx, dx)
z = arange(center, center+zpoints*dx, dx)
Z, R = meshgrid(z, r)
gmax = g3mc.max()
xplot.set_xlim(xlow, xhigh)
zplot.set_xlim(x_to_z(xlow), x_to_z(xhigh))
xticks = [-6, -4, -2, 0]
zticks = [4, 6, 8, 10]
plotticks = xticks + [z_to_x(z) for z in zticks]
xplot.set_xticks(plotticks)
xplot.set_xticklabels(['$%i$' %tick for tick in xticks[:-1]] + ['$0$ $2$'] +
['$%i$' %tick for tick in zticks])
zplot.set_xticks([])
zplot.axvline(x=rpath, color='k')
zplot.axvline(x=3*rpath, color='k')
bracket.bracket(xplot, -0.01, -xlow/(xhigh - xlow), -.06, .06, r'$x/R$')
bracket.bracket(xplot, -xlow/(xhigh - xlow), 1.01, -.06, .06, r'$z/R$')
xmin = 1.0
xmax = 9.0
ymax = 6.0
twod_plot.set_xlim(zmin, zmax)
twod_plot.set_ylim(-rmax, rmax)
#fig.subplots_adjust(hspace=0.001)
twod_plot.set_aspect('equal')
g3mc = read_triplet(ff, 'mc')[:, center/dx:-1]
rpoints = len(g3mc[:,0])
zpoints = len(g3mc[0,:])
xplot.set_xlim(xlow, xhigh)
zplot.set_xlim(x_to_z(xlow), x_to_z(xhigh))
xticks = [-6, -4, -2, 0]
zticks = [4, 6, 8, 10]
plotticks = xticks + [z_to_x(z) for z in zticks]
xplot.set_xticks(plotticks)
xplot.set_xticklabels(['$%i$' % tick for tick in xticks[:-1]] + ['$0$ $2$'] +
['$%i$' % tick for tick in zticks])
zplot.set_xticks([])
zplot.axvline(x=rpath, color='k')
zplot.axvline(x=3 * rpath, color='k')
bracket.bracket(xplot, -0.01, -xlow / (xhigh - xlow), -.06, .06, r'$x/R$')
bracket.bracket(xplot, -xlow / (xhigh - xlow), 1.01, -.06, .06, r'$z/R$')
xmin = 1.0
xmax = 9.0
ymax = 6.0
twod_plot.set_xlim(zmin, zmax)
twod_plot.set_ylim(-rmax, rmax)
#fig.subplots_adjust(hspace=0.001)
twod_plot.set_aspect('equal')
g3mc = read_triplet(ff, 'mc')[:, int(center / dx):-1]
rpoints = len(g3mc[:, 0])
zpoints = len(g3mc[0, :])
