def test__maybe_make_grouper_single_groupby(self):
from cupcake.smush.base import SmushPlotterBase
p = SmushPlotterBase()
p.establish_variables(self.data)
test_grouper = p._maybe_make_grouper('o', None, str)
true_grouper = pd.Series(['o']*self.matrix.shape[0],
index=self.data.index)
pdt.assert_series_equal(test_grouper, true_grouper)
def test__maybe_make_grouper_single_groupby(self):
from cupcake.smush.base import SmushPlotterBase
p = SmushPlotterBase()
p.establish_variables(self.data)
test_grouper = p._maybe_make_grouper('o', None, str)
true_grouper = pd.Series(['o'] * self.matrix.shape[0],
index=self.data.index)
pdt.assert_series_equal(test_grouper, true_grouper)
def test__maybe_make_grouper_multiple_groupbys(self):
from cupcake.smush.base import SmushPlotterBase
p = SmushPlotterBase()
p.establish_variables(self.data)
half = int(self.nrow/2.)
groupby = pd.Series((['B'] * half) + (['A'] * half))
order = ['A', 'B']
test_grouper = p._maybe_make_grouper(groupby, order, str)
true_grouper = pd.Series(pd.Categorical(groupby, categories=order,
ordered=True), index=self.data.index)
pdt.assert_series_equal(test_grouper, true_grouper)
def test__maybe_make_grouper_multiple_groupbys(self):
from cupcake.smush.base import SmushPlotterBase
p = SmushPlotterBase()
p.establish_variables(self.data)
half = int(self.nrow / 2.)
groupby = pd.Series((['B'] * half) + (['A'] * half))
order = ['A', 'B']
test_grouper = p._maybe_make_grouper(groupby, order, str)
true_grouper = pd.Series(pd.Categorical(groupby,
categories=order,
ordered=True),
index=self.data.index)
pdt.assert_series_equal(test_grouper, true_grouper)