def test_wrong_bases():
# x axis not transformed
p = (ggplot(df, aes('x', 'x')) +
annotation_logticks(sides='b', size=.75, base=10) + geom_point())
with pytest.warns(PlotnineWarning):
p.draw_test()
# x axis not transform, but ticks requested for a different base
p = (ggplot(df, aes('x', 'x')) +
annotation_logticks(sides='b', size=.75, base=10) +
scale_x_continuous(trans=log_trans(8)) + geom_point())
with pytest.warns(PlotnineWarning):
p.draw_test()
# x axis is discrete
df2 = df.assign(discrete=pd.Categorical([str(a) for a in df['x']]))
p = (ggplot(df2, aes('discrete', 'x')) +
annotation_logticks(sides='b', size=.75, base=None) + geom_point())
with pytest.warns(PlotnineWarning):
p.draw_test()
# y axis is discrete
df2 = df.assign(discrete=pd.Categorical([str(a) for a in df['x']]))
p = (ggplot(df2, aes('x', 'discrete')) +
annotation_logticks(sides='l', size=.75, base=None) + geom_point())
with pytest.warns(PlotnineWarning):
p.draw_test()
def test_annotation_logticks_coord_flip():
p = (ggplot(df, aes('x', 'x')) + annotation_logticks(sides='b', size=.75) +
geom_point() + scale_x_log10() + scale_y_log10() + coord_flip() +
theme(panel_grid_minor=element_line(color='green'),
panel_grid_major=element_line(color='red')))
assert p == 'annotation_logticks_coord_flip'
def test_annotation_logticks():
# The grid should align with the logticks
p = (ggplot(df, aes('x', 'x')) + annotation_logticks(sides='b', size=.75) +
geom_point() + scale_x_log10() + scale_y_log10() +
theme(panel_grid_minor=element_line(color='green'),
panel_grid_major=element_line(color='red')))
assert p == 'annotation_logticks'
def test_annotation_logticks_coord_flip_discrete_bottom():
df2 = df.assign(discrete=pd.Categorical(['A' + str(a) for a in df['x']]))
p = (ggplot(df2, aes('x', 'discrete')) +
annotation_logticks(sides='b', size=.75) + geom_point() +
scale_x_log10() + coord_flip() +
theme(panel_grid_minor=element_line(color='green'),
panel_grid_major=element_line(color='red')))
assert p == 'annotation_logticks_coord_flip_discrete_bottom'
def test_annotation_logticks_base_8():
base = 8
df = pd.DataFrame({'x': base**np.arange(4)})
# The grid should align with the logticks
p = (ggplot(df, aes('x', 'x')) + annotation_logticks(sides='b', size=.75) +
geom_point() + scale_x_continuous(trans=log_trans(base=base)) +
theme(panel_grid_minor=element_line(color='green'),
panel_grid_major=element_line(color='red')))
assert p == 'annotation_logticks_base_8'
def test_wrong_bases():
# x axis not transformed
p = (ggplot(df, aes('x', 'x'))
+ annotation_logticks(sides='b', size=.75, base=10)
+ geom_point()
)
with pytest.warns(PlotnineWarning):
p.draw_test()
# x axis not transform, but ticks requested for a different base
p = (ggplot(df, aes('x', 'x'))
+ annotation_logticks(sides='b', size=.75, base=10)
+ scale_x_continuous(trans=log_trans(8))
+ geom_point()
)
with pytest.warns(PlotnineWarning):
p.draw_test()
def test_annotation_logticks_faceting():
n = len(df)
df2 = pd.DataFrame({
'x': np.hstack([df['x'], df['x']]),
'g': list('a' * n + 'b' * n)
})
p = (ggplot(df2) + annotation_logticks(sides='b', size=.75) +
geom_point(aes('x', 'x')) + scale_x_log10() + scale_y_log10() +
facet_wrap('g') + theme(panel_grid_minor=element_line(color='green'),
panel_grid_major=element_line(color='red')))
assert p == 'annotation_logticks_faceting'
