def test_establish_symbols_text_true(self):
from cupcake.smush.base import SmushPlotterBase
symbol_kws = self.symbol_kws.copy()
symbol_kws['text'] = True
p = SmushPlotterBase()
p.establish_variables(self.data)
p.establish_symbols(**symbol_kws)
pdt.assert_series_equal(p.symbol,
pd.Series(map(str, list(range(self.nrow))),
index=self.data.index))
assert p.text
def test_establish_symbols_defaults(self):
from cupcake.smush.base import SmushPlotterBase
p = SmushPlotterBase()
p.establish_variables(self.data)
p.establish_symbols(**self.symbol_kws)
pdt.assert_series_equal(p.symbol, pd.Series(['o']*self.nrow,
index=self.data.index))
pdt.assert_series_equal(p.linewidth, pd.Series([1]*self.nrow,
index=self.data.index))
pdt.assert_series_equal(p.edgecolor, pd.Series(['k']*self.nrow,
index=self.data.index))
assert not p.text
def test_establish_symbols_text_true(self):
from cupcake.smush.base import SmushPlotterBase
symbol_kws = self.symbol_kws.copy()
symbol_kws['text'] = True
p = SmushPlotterBase()
p.establish_variables(self.data)
p.establish_symbols(**symbol_kws)
pdt.assert_series_equal(
p.symbol,
pd.Series(map(str, list(range(self.nrow))), index=self.data.index))
assert p.text
def test_establish_symbols_defaults(self):
from cupcake.smush.base import SmushPlotterBase
p = SmushPlotterBase()
p.establish_variables(self.data)
p.establish_symbols(**self.symbol_kws)
pdt.assert_series_equal(
p.symbol, pd.Series(['o'] * self.nrow, index=self.data.index))
pdt.assert_series_equal(
p.linewidth, pd.Series([1] * self.nrow, index=self.data.index))
pdt.assert_series_equal(
p.edgecolor, pd.Series(['k'] * self.nrow, index=self.data.index))
assert not p.text
def test_establish_symbols_text_series_ordered(self):
from cupcake.smush.base import SmushPlotterBase
symbol_kws = self.symbol_kws.copy()
symbol_kws['text'] = self.groupby
symbol_kws['text_order'] = self.order
p = SmushPlotterBase()
p.establish_variables(self.data)
p.establish_symbols(**symbol_kws)
symbol = pd.Series(pd.Categorical(self.groupby, ordered=True,
categories=self.order), index=self.data.index)
pdt.assert_series_equal(p.symbol, symbol)
assert p.text
def test_establish_symbols_text_series_not_str(self):
from cupcake.smush.base import SmushPlotterBase
half = int(self.nrow/2.)
text = pd.Series(([1] * half) + ([2] * half)).map(str)
symbol_kws = self.symbol_kws.copy()
symbol_kws['text'] = text
p = SmushPlotterBase()
p.establish_variables(self.data)
p.establish_symbols(**symbol_kws)
symbol = pd.Series(pd.Categorical(text, ordered=True,
categories=['1', '2']), index=self.data.index)
pdt.assert_series_equal(p.symbol, symbol)
assert p.text
def test_establish_symbols_text_series_ordered(self):
from cupcake.smush.base import SmushPlotterBase
symbol_kws = self.symbol_kws.copy()
symbol_kws['text'] = self.groupby
symbol_kws['text_order'] = self.order
p = SmushPlotterBase()
p.establish_variables(self.data)
p.establish_symbols(**symbol_kws)
symbol = pd.Series(pd.Categorical(self.groupby,
ordered=True,
categories=self.order),
index=self.data.index)
pdt.assert_series_equal(p.symbol, symbol)
assert p.text
def test_establish_symbols_text_series_not_str(self):
from cupcake.smush.base import SmushPlotterBase
half = int(self.nrow / 2.)
text = pd.Series(([1] * half) + ([2] * half)).map(str)
symbol_kws = self.symbol_kws.copy()
symbol_kws['text'] = text
p = SmushPlotterBase()
p.establish_variables(self.data)
p.establish_symbols(**symbol_kws)
symbol = pd.Series(pd.Categorical(text,
ordered=True,
categories=['1', '2']),
index=self.data.index)
pdt.assert_series_equal(p.symbol, symbol)
assert p.text
