def test11(self):
bg = True
n = 4
self.M = EzTemplate.Multi(
rows=1,
columns=1,
legend_direction='vertical',
right_margin=.2,
left_margin=.15,
top_margin=.15,
bottom_margin=.15)
t = self.M.get()
leg = t.legend
tt = self.x.gettexttable(leg.texttable)
to = self.x.gettextorientation(leg.textorientation)
OD = EzTemplate.oneD(n=n, template=t)
x = MV2.arange(0, 360)
x = x / 180. * MV2.pi
ax = cdms2.createAxis(x)
self.x.portrait()
for i in range(n):
y = MV2.sin((i + 1) * x)
y.setAxis(0, ax)
yx = self.x.createyxvsx()
yx.linecolor = 241 + i
yx.datawc_y1 = -1.
yx.datawc_y2 = 1.
y.id = str(i)
t = OD.get()
self.x.plot(y, t, yx, bg=bg)
self.checkImage('test_EzTemplate_11.png')
def testMultiple(self):
f = cdms2.open(os.path.join(cdat_info.get_sampledata_path(), 'clt.nc'))
s = f('clt', slice(0, 1))
f.close()
M = EzTemplate.Multi(rows=2, columns=2)
for i in range(4):
self.x.plot(s, M.get())
self.checkImage("test_vcsaddons_multiple_x.png")
y = vcs.init(bg=True, geometry={"width": 1200, "height": 1090})
M2 = EzTemplate.Multi(rows=1, columns=2)
for i in range(2):
y.plot(s, M2.get())
y.png("tests_png/test_vcsaddons_multiple_y.png")
self.checkImage("test_vcsaddons_multiple_y.png", pngReady=True)
def setUp(self):
# This is for circleci that crashes for any mac bg=True
self.x = vcs.init(geometry=self.geometry, bg=self.bg)
self.x.setantialiasing(0)
self.x.drawlogooff()
if self.geometry is not None:
self.x.setbgoutputdimensions(self.geometry['width'],
self.geometry['height'],
units="pixels")
# if not self.bg:
# self.x.open()
self.orig_cwd = os.getcwd()
self.pngsdir = "tests_png"
if not os.path.exists(self.pngsdir):
os.makedirs(self.pngsdir)
self.basedir = os.path.join("tests", "baselines")
self.basedatadir = os.path.join("tests", "data")
self.M = EzTemplate.Multi(rows=4, columns=3, x=self.x)
import os, sys, vcs.testing.regression as regression
import vcs
from vcsaddons import EzTemplate
import cdms2, vcs, sys
## 12 plots 1 legend per row on the right
## Initialize VCS
x = vcs.init()
x.drawlogooff()
bg = True
M = EzTemplate.Multi(rows=4, columns=3)
M.legend.direction = 'vertical'
for i in range(12):
t = M.get(legend='local')
if i % 3 != 2:
t.legend.priority = 0 # Turn off legend
fnm = "test_12_plot_one_leg_per_row_right.png"
M.preview(fnm, bg=bg)
ret = regression.check_result_image(fnm, sys.argv[1])
if not bg:
raw_input("Press Enter")
sys.exit(ret)