def test_xy_limits(): lst = [1, 2, 3] arr = np.array(lst) series = pd.Series(lst) s1 = scale_x_discrete(limits=lst) s2 = scale_x_discrete(limits=arr) s3 = scale_x_discrete(limits=series) assert all(s2.limits == s1.limits) assert all(s3.limits == s1.limits)
def test_discrete_xy_scale_limits(): lst = list('abcd') x = pd.Series(pd.Categorical(lst, ordered=True)) s = scale_x_discrete() s.train(x) assert s.limits == lst s = scale_x_discrete(limits=reversed) s.train(x) assert s.limits == lst[::-1]
def test_discrete_xy_scale_drop_limits(): df = pd.DataFrame({ 'x': list('aaaabbbbccccddd'), 'c': list('112312231233123') }) p = (ggplot(df) + geom_bar(aes(x='x', fill='c')) + scale_x_discrete(limits=list('abc'))) assert p == 'discrete_xy_scale_drop_limits'
def test_xy_palette(): s = scale_x_discrete() value = s.palette(3) assert (value == 3) s = scale_y_discrete() value = s.palette(11.5) assert (value == 11.5) s = scale_x_continuous() value = s.palette(3.63) assert (value == 3.63) s = scale_y_continuous() value = s.palette(11.52) assert (value == 11.52)