def setUpClass(cls): cls.ds = open_dataset(cls.ncfile) cls.data = InteractiveList.from_dataset( cls.ds, y=0, z=0, t=0, name=cls.var, auto_update=True) cls.data[0].arr_name = 'arr0' cls.data.arr_name = 'arr0' cls.plotter = BarPlotter(cls.data[0]) cls.create_dirs()
def setUpClass(cls): cls.ds = open_dataset(cls.ncfile) cls.data = InteractiveList.from_dataset(cls.ds, y=[0, 1], z=0, t=0, name=cls.var, auto_update=True) cls.plotter = BasePlotter(cls.data)
def _test_DtTicksBase(self, *args): data = InteractiveList.from_dataset( self.data[0].base, y=[0, 1], z=0, x=0, name=self.var, auto_update=True) plotter = self.plotter.__class__(data) ax = plotter.ax plotter.update(xticklabels='%m') self.assertListEqual(ax.get_xticks().astype(int).tolist(), list(range(5)))
def _test_DtTicksBase(self, *args): # testing of psyplot.plotter.simple.DtTicksBase args = iter(args) data = InteractiveList.from_dataset( self.data[0].base, y=[0, 1], z=0, x=0, name=self.var, auto_update=True) plotter = self.plotter.__class__(data) ax = plotter.ax xticks = {'major': ax.get_xticks(), 'minor': ax.get_xticks(minor=True)} plotter.update(xticks='month') self.assertEqual(list(plt.gca().get_xticks()), [722494.75, 722524.25, 722554.75, 722585.25]) plotter.update(xticks='monthbegin') self.assertEqual( list(plt.gca().get_xticks()), [722450.75, 722481.75, 722509.75, 722540.75, 722570.75, 722601.75]) plotter.update(xticks='monthend') self.assertEqual( list(plt.gca().get_xticks()), [722480.75, 722508.75, 722539.75, 722569.75, 722600.75]) plotter.update(xticks='month', xticklabels='%m') # sometimes the labels are only set after drawing if ax.get_xticklabels()[0].get_text(): self.assertEqual( [int(t.get_text()) for t in ax.get_xticklabels()[:]], list(range(2, 6))) plotter.update(xticks={'minor': 'week'}, xticklabels={'minor': '%d'}, xtickprops={'pad': 7.0}) self.assertEqual( plotter.ax.get_xticks(minor=True).tolist(), [722487.75, 722494.75, 722501.75, 722508.75, 722515.75, 722522.75, 722529.75, 722536.75, 722543.75, 722550.75, 722557.75, 722564.75, 722571.75, 722578.75, 722585.75, 722592.75, 722599.75]) self.compare_figures(next(args, self.get_ref_file('xticks'))) plotter.update(xticks={'major': None, 'minor': None}) self.assertEqual(list(plotter.ax.get_xticks()), list(xticks['major'])) self.assertEqual(list(plotter.ax.get_xticks(minor=True)), list(xticks['minor']))
def setUpClass(cls): cls.ds = open_dataset(cls.ncfile) cls.data = InteractiveList.from_dataset( cls.ds, y=[0, 1], z=0, t=0, name=cls.var, auto_update=True) cls.plotter = BasePlotter(cls.data)