def test_two_plots_with_independent_axes(self):
c1 = Contour(self.cube, self.axes)
levels = 5
other = _add_levels(self.cube, levels)[:, 0]
c2 = Contour(other, self.axes)
browser = Browser([c1, c2])
tcoord = self._tidy(self.cube.coord('time'))
zcoord = self._tidy(other.coord('model_level_number'))
# Check axis_by_name.
taxis = _AxisDefn(dim=0, name='time', size=7, coord=tcoord)
zaxis = _AxisDefn(dim=0,
name='model_level_number',
size=levels,
coord=zcoord)
axis_by_name = browser._axis_by_name
expected = dict(time=taxis, model_level_number=zaxis)
self.assertEqual(axis_by_name, expected)
# Check plots_by_name.
c1_id, c2_id = id(c1), id(c2)
plots_by_name = browser._plots_by_name
self.assertEqual(set(plots_by_name.keys()),
set(['time', 'model_level_number']))
self.assertEqual([id(v) for v in plots_by_name['time']], [c1_id])
self.assertEqual([id(v) for v in plots_by_name['model_level_number']],
[c2_id])
# Check names_by_plot_id.
names_by_plot_id = browser._names_by_plot_id
self.assertEqual(set(names_by_plot_id.keys()), set([c1_id, c2_id]))
self.assertEqual(names_by_plot_id[c1_id], ['time'])
self.assertEqual(names_by_plot_id[c2_id], ['model_level_number'])
def test_two_plots_with_independent_axes(self):
c1 = Contour(self.cube, self.axes)
levels = 5
other = _add_levels(self.cube, levels)[:, 0]
c2 = Contour(other, self.axes)
browser = Browser([c1, c2])
tcoord = self._tidy(self.cube.coord('time'))
zcoord = self._tidy(other.coord('model_level_number'))
# Check axis_by_name.
taxis = _AxisDefn(dim=0, name='time', size=7, coord=tcoord)
zaxis = _AxisDefn(dim=0, name='model_level_number', size=levels,
coord=zcoord)
axis_by_name = browser._axis_by_name
expected = dict(time=taxis, model_level_number=zaxis)
self.assertEqual(axis_by_name, expected)
# Check plots_by_name.
c1_id, c2_id = id(c1), id(c2)
plots_by_name = browser._plots_by_name
self.assertEqual(set(plots_by_name.keys()),
set(['time', 'model_level_number']))
self.assertEqual([id(v) for v in plots_by_name['time']],
[c1_id])
self.assertEqual([id(v) for v in plots_by_name['model_level_number']],
[c2_id])
# Check names_by_plot_id.
names_by_plot_id = browser._names_by_plot_id
self.assertEqual(set(names_by_plot_id.keys()), set([c1_id, c2_id]))
self.assertEqual(names_by_plot_id[c1_id], ['time'])
self.assertEqual(names_by_plot_id[c2_id], ['model_level_number'])
def test_lookup(self):
dim, name, size, coord = 0, 1, 2, 4
a = _AxisDefn(dim=dim, name=name, size=size, coord=coord)
self.assertEqual(a.dim, dim)
self.assertEqual(a.name, name)
self.assertEqual(a.size, size)
self.assertEqual(a.coord, coord)
def test_slider_grid_latitude(self):
coords = ('time', 'grid_longitude')
plot = Plot2D(self.cube, self.axes, coords=coords)
original = self.cube.coord('grid_latitude')
coord = self._tidy(original)
expected = _AxisDefn(dim=1, name='grid_latitude', size=9, coord=coord)
actual = plot.sliders_axis
self.assertEqual(actual, [expected])
self.assertIsNot(actual[0].coord, original)
def test_slider_grid_latitude(self):
coords = ('time', 'grid_longitude')
plot = Plot2D(self.cube, self.axes, coords=coords)
original = self.cube.coord('grid_latitude')
coord = self._tidy(original)
expected = _AxisDefn(dim=1, name='grid_latitude', size=9, coord=coord)
actual = plot.sliders_axis
self.assertEqual(actual, [expected])
self.assertIsNot(actual[0].coord, original)
def test_single_plot_with_axis(self):
plot = Contour(self.cube, self.axes)
browser = Browser(plot)
name = 'time'
coord = self._tidy(self.cube.coord(name))
# Check axis_by_name.
axis = _AxisDefn(dim=0, name=name, size=7, coord=coord)
expected = dict(time=axis)
self.assertEqual(browser._axis_by_name, expected)
# Check plots_by_name.
plot_id = id(plot)
self.assertEqual(browser._plots_by_name.keys(), [name])
self.assertEqual([id(v) for v in browser._plots_by_name[name]],
[plot_id])
# Check names_by_plot_id.
self.assertEqual(browser._names_by_plot_id.keys(), [plot_id])
self.assertEqual(browser._names_by_plot_id[plot_id], [name])
def test_single_plot_with_axis(self):
plot = Contour(self.cube, self.axes)
browser = Browser(plot)
name = 'time'
coord = self._tidy(self.cube.coord(name))
# Check axis_by_name.
axis = _AxisDefn(dim=0, name=name, size=7, coord=coord)
expected = dict(time=axis)
self.assertEqual(browser._axis_by_name, expected)
# Check plots_by_name.
plot_id = id(plot)
self.assertEqual(browser._plots_by_name.keys(), [name])
self.assertEqual([id(v) for v in browser._plots_by_name[name]],
[plot_id])
# Check names_by_plot_id.
self.assertEqual(browser._names_by_plot_id.keys(), [plot_id])
self.assertEqual(browser._names_by_plot_id[plot_id], [name])
def test_two_plots_with_shared_axis(self):
coords = ('time', 'grid_latitude')
c1 = Contour(self.cube, self.axes, coords=coords)
c2 = Contour(self.cube, self.axes, coords=coords)
browser = Browser([c1, c2])
name = 'grid_longitude'
coord = self._tidy(self.cube.coord(name))
# Check axis_by_name.
axis = _AxisDefn(dim=2, name=name, size=11, coord=coord)
expected = dict(grid_longitude=axis)
self.assertEqual(browser._axis_by_name, expected)
# Check plots_by_name.
c1_id, c2_id = id(c1), id(c2)
plots_by_name = browser._plots_by_name
self.assertEqual(plots_by_name.keys(), [name])
self.assertEqual(set([id(v) for v in plots_by_name[name]]),
set([c1_id, c2_id]))
# Check names_by_plot_id.
names_by_plot_id = browser._names_by_plot_id
self.assertEqual(set(names_by_plot_id.keys()), set([c1_id, c2_id]))
self.assertEqual(names_by_plot_id[c1_id], [name])
self.assertEqual(names_by_plot_id[c2_id], [name])
def test_two_plots_with_shared_axis(self):
coords = ('time', 'grid_latitude')
c1 = Contour(self.cube, self.axes, coords=coords)
c2 = Contour(self.cube, self.axes, coords=coords)
browser = Browser([c1, c2])
name = 'grid_longitude'
coord = self._tidy(self.cube.coord(name))
# Check axis_by_name.
axis = _AxisDefn(dim=2, name=name, size=11, coord=coord)
expected = dict(grid_longitude=axis)
self.assertEqual(browser._axis_by_name, expected)
# Check plots_by_name.
c1_id, c2_id = id(c1), id(c2)
plots_by_name = browser._plots_by_name
self.assertEqual(plots_by_name.keys(), [name])
self.assertEqual(set([id(v) for v in plots_by_name[name]]),
set([c1_id, c2_id]))
# Check names_by_plot_id.
names_by_plot_id = browser._names_by_plot_id
self.assertEqual(set(names_by_plot_id.keys()), set([c1_id, c2_id]))
self.assertEqual(names_by_plot_id[c1_id], [name])
self.assertEqual(names_by_plot_id[c2_id], [name])
def test_alias_with_defn(self):
a1 = _AxisAlias(dim=0, name=1, size=2)
a2 = _AxisDefn(dim=0, name=1, size=2, coord=3)
self.assertNotEqual(a1, a2)
def test_defn_with_different_coord(self):
a1 = _AxisDefn(dim=0, name=1, size=2, coord=3)
a2 = _AxisDefn(dim=0, name=1, size=2, coord=30)
self.assertNotEqual(a1, a2)
def test_defn_same(self):
a1 = _AxisDefn(dim=0, name=1, size=2, coord=3)
a2 = _AxisDefn(dim=0, name=1, size=2, coord=3)
self.assertEqual(a1, a2)