def test_layersvisitor_03(): '''Nested Layers work as expected.''' r1 = rectangle(1000, 1000) r2 = rectangle(500, 500) t3 = isosceles(500, 200) l1 = Layer([r1], 1) l2 = Layer([l1, r2], 2) l3 = Layer([t3, l2], 3) v = LayersVisitor() v.visit(l3) assert v.layers.keys() == [1, 2, 3] assert v.layers[1] == [r1] assert v.layers[2] == [r2] assert v.layers[3] == [t3]
def test_layersvisitor_02(): '''One paths with layer gorup.''' r1 = rectangle(1000, 1000) l = Layer([r1], 1) v = LayersVisitor() v.visit(l) assert v.layers.keys() == [1] assert v.layers[1] == [r1]
def test_layervisitor_05(): """Multiple nestings work as expected.""" c1, c2, c3 = [circle(1000 * i) for i in range(1, 4)] r = rectangle(1000, 1000) t = isosceles(1000, 2000) l_leaf = Layer([c1], "leaf") g1 = Group([l_leaf, c2, c3]) l_bottom = Layer([g1], "bottom") l_med = Layer([l_bottom, r], "med") l_top = Layer([l_med, t], "top") v = LayersVisitor() v.visit(l_top) assert sorted(v.layers.keys()) == ["bottom", "leaf", "med", "top"] assert v.layers["top"] == [t] assert v.layers["med"] == [r] assert v.layers["bottom"] == [c2, c3] assert v.layers["leaf"] == [c1]
def test_layervisitor_05(): '''Multiple nestings work as expected.''' c1, c2, c3 = [circle(1000 * i) for i in range(1, 4)] r = rectangle(1000, 1000) t = isosceles(1000, 2000) l_leaf = Layer([c1], 'leaf') g1 = Group([l_leaf, c2, c3]) l_bottom = Layer([g1], 'bottom') l_med = Layer([l_bottom, r], 'med') l_top = Layer([l_med, t], 'top') v = LayersVisitor() v.visit(l_top) assert sorted(v.layers.keys()) == ['bottom', 'leaf', 'med', 'top'] assert v.layers['top'] == [t] assert v.layers['med'] == [r] assert v.layers['bottom'] == [c2, c3] assert v.layers['leaf'] == [c1]
def layer(shapes, name): return Layer(shapes, name)