def test_addition(self):
p = ggplot(df, aes('x', 'y'))
p1 = p + self.lyrs[0] + self.lyrs[1] + self.lyrs[2]
assert _get_colors(p1) == colors
p2 = p + self.lyrs
assert _get_colors(p2) == colors
# Real layers
lyrs = Layers(layer.from_geom(obj) for obj in self.lyrs)
p3 = p + lyrs
assert _get_colors(p3) == colors
p += self.lyrs
assert _get_colors(p) == colors
with pytest.raises(PlotnineError):
geom_point() + layer.from_geom(geom_point())
with pytest.raises(PlotnineError):
geom_point() + self.lyrs
def test_addition(self):
p = ggplot(df, aes('x', 'y'))
p1 = p + self.lyrs[0] + self.lyrs[1] + self.lyrs[2]
assert _get_colors(p1) == colors
p2 = p + self.lyrs
assert _get_colors(p2) == colors
# Real layers
lyrs = Layers(layer.from_geom(obj) for obj in self.lyrs)
p3 = p + lyrs
assert _get_colors(p3) == colors
p += self.lyrs
assert _get_colors(p) == colors
with pytest.raises(PlotnineError):
geom_point() + layer.from_geom(geom_point())
with pytest.raises(PlotnineError):
geom_point() + self.lyrs