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)
