Skip to content

Commit

Permalink
Add test for colored drawing with frames
Browse files Browse the repository at this point in the history
  • Loading branch information
@neiljdo
neiljdo committed Nov 7, 2024
1 parent 9e2aa45 commit 8af37b1
Showing 1 changed file with 220 additions and 0 deletions.
220 changes: 220 additions & 0 deletions tests/backend/test_frame_drawing.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -377,6 +377,217 @@
]


colored_tikz_outputs = [
"""\\begin{tikzpicture}[baseline=(0.base)]
\\begin{pgfonlayer}{nodelayer}
\\node (0) at (0, 0) {};
\\node [] (1) at (2.25, 11.625) {};
\\node [] (2) at (3.25, 11.625) {};
\\node [] (3) at (3.25, 12.125) {};
\\node [] (4) at (2.25, 12.125) {};
\\node [style=none, fill=white] (5) at (2.75, 11.875) {Alice};
\\node [] (6) at (4.75, 11.625) {};
\\node [] (7) at (5.75, 11.625) {};
\\node [] (8) at (5.75, 12.125) {};
\\node [] (9) at (4.75, 12.125) {};
\\node [style=none, fill=white] (10) at (5.25, 11.875) {Bob};
\\node [] (11) at (7.25, 11.625) {};
\\node [] (12) at (8.25, 11.625) {};
\\node [] (13) at (8.25, 12.125) {};
\\node [] (14) at (7.25, 12.125) {};
\\node [style=none, fill=white] (15) at (7.75, 11.875) {cake};
\\node [] (16) at (9.75, 11.625) {};
\\node [] (17) at (10.75, 11.625) {};
\\node [] (18) at (10.75, 12.125) {};
\\node [] (19) at (9.75, 12.125) {};
\\node [style=none, fill=white] (20) at (10.25, 11.875) {coffee};
\\node [] (21) at (2.25, 9.125) {};
\\node [] (22) at (5.75, 9.125) {};
\\node [] (23) at (5.75, 9.625) {};
\\node [] (24) at (2.25, 9.625) {};
\\node [style=none, fill=white] (25) at (4.0, 9.375) {told};
\\node [] (26) at (-1.5, -1.875) {};
\\node [] (27) at (14.5, -1.875) {};
\\node [] (28) at (14.5, 7.125) {};
\\node [] (29) at (-1.5, 7.125) {};
\\node [style=none, fill=white] (30) at (6.5, 6.625) {and};
\\node [] (31) at (-1.0, 1.625) {};
\\node [] (32) at (3.5, 1.625) {};
\\node [] (33) at (3.5, 3.625) {};
\\node [] (34) at (-1.0, 3.625) {};
\\node [] (35) at (-0.5, 2.375) {};
\\node [] (36) at (3.0, 2.375) {};
\\node [] (37) at (3.0, 2.875) {};
\\node [] (38) at (-0.5, 2.875) {};
\\node [style=none, fill=white] (39) at (1.25, 2.625) {eats};
\\node [] (40) at (4.25, 1.625) {};
\\node [] (41) at (8.75, 1.625) {};
\\node [] (42) at (8.75, 3.625) {};
\\node [] (43) at (4.25, 3.625) {};
\\node [] (44) at (4.75, 2.375) {};
\\node [] (45) at (8.25, 2.375) {};
\\node [] (46) at (8.25, 2.875) {};
\\node [] (47) at (4.75, 2.875) {};
\\node [style=none, fill=white] (48) at (6.5, 2.625) {drinks};
\\node [] (49) at (9.5, -0.875) {};
\\node [] (50) at (14.0, -0.875) {};
\\node [] (51) at (14.0, 6.125) {};
\\node [] (52) at (9.5, 6.125) {};
\\node [] (53) at (11.25, 4.875) {};
\\node [] (54) at (12.25, 4.875) {};
\\node [] (55) at (12.25, 5.375) {};
\\node [] (56) at (11.25, 5.375) {};
\\node [style=none, fill=white] (57) at (11.75, 5.125) {runs};
\\node [] (58) at (10.0, 2.375) {};
\\node [] (59) at (13.5, 2.375) {};
\\node [] (60) at (13.5, 2.875) {};
\\node [] (61) at (10.0, 2.875) {};
\\node [style=none, fill=white] (62) at (11.75, 2.625) {x};
\\node [] (63) at (10.0, -0.125) {};
\\node [] (64) at (11.0, -0.125) {};
\\node [] (65) at (11.0, 0.375) {};
\\node [] (66) at (10.0, 0.375) {};
\\node [style=none, fill=white] (67) at (10.5, 0.125) {y};
\\node [] (68) at (6.0, -4.375) {};
\\node [] (69) at (9.5, -4.375) {};
\\node [] (70) at (9.5, -3.875) {};
\\node [] (71) at (6.0, -3.875) {};
\\node [style=none, fill=white] (72) at (7.75, -4.125) {test};
\\node [] (73) at (3.5, -4.375) {};
\\node [] (74) at (4.5, -4.375) {};
\\node [] (75) at (4.5, -3.875) {};
\\node [] (76) at (3.5, -3.875) {};
\\node [style=none, fill=white] (77) at (4.0, -4.125) {a};
\\node [] (78) at (8.5, -6.875) {};
\\node [] (79) at (9.5, -6.875) {};
\\node [] (80) at (9.5, -6.375) {};
\\node [] (81) at (8.5, -6.375) {};
\\node [style=none, fill=white] (82) at (9.0, -6.625) {b};
\\node [] (83) at (2.75, 11.625) {};
\\node [] (84) at (2.75, 9.625) {};
\\node [style=none, fill=white, right] (85) at (2.85, 11.525) {n};
\\node [] (86) at (5.25, 11.625) {};
\\node [] (87) at (5.25, 9.625) {};
\\node [style=none, fill=white, right] (88) at (5.35, 11.525) {n};
\\node [] (89) at (2.75, 9.125) {};
\\node [] (90) at (2.75, 7.125) {};
\\node [style=none, fill=white, right] (91) at (2.85, 9.025) {n};
\\node [] (92) at (5.25, 9.125) {};
\\node [] (93) at (5.25, 7.125) {};
\\node [style=none, fill=white, right] (94) at (5.35, 9.025) {n};
\\node [] (95) at (7.75, 11.625) {};
\\node [] (96) at (7.75, 7.125) {};
\\node [style=none, fill=white, right] (97) at (7.85, 11.525) {n};
\\node [] (98) at (10.25, 11.625) {};
\\node [] (99) at (10.25, 7.125) {};
\\node [style=none, fill=white, right] (100) at (10.35, 11.525) {n};
\\node [] (101) at (0.0, 3.625) {};
\\node [] (102) at (0.0, 2.875) {};
\\node [style=none, fill=white, right] (103) at (0.1, 3.625) {n};
\\node [] (104) at (2.5, 3.625) {};
\\node [] (105) at (2.5, 2.875) {};
\\node [style=none, fill=white, right] (106) at (2.6, 3.625) {n};
\\node [] (107) at (0.0, 2.375) {};
\\node [] (108) at (0.0, 1.625) {};
\\node [style=none, fill=white, right] (109) at (0.1, 2.275) {n};
\\node [] (110) at (2.5, 2.375) {};
\\node [] (111) at (2.5, 1.625) {};
\\node [style=none, fill=white, right] (112) at (2.6, 2.275) {n};
\\node [] (113) at (5.25, 3.625) {};
\\node [] (114) at (5.25, 2.875) {};
\\node [style=none, fill=white, right] (115) at (5.35, 3.625) {n};
\\node [] (116) at (7.75, 3.625) {};
\\node [] (117) at (7.75, 2.875) {};
\\node [style=none, fill=white, right] (118) at (7.85, 3.625) {n};
\\node [] (119) at (5.25, 2.375) {};
\\node [] (120) at (5.25, 1.625) {};
\\node [style=none, fill=white, right] (121) at (5.35, 2.275) {n};
\\node [] (122) at (7.75, 2.375) {};
\\node [] (123) at (7.75, 1.625) {};
\\node [style=none, fill=white, right] (124) at (7.85, 2.275) {n};
\\node [] (125) at (11.75, 6.125) {};
\\node [] (126) at (11.75, 5.375) {};
\\node [style=none, fill=white, right] (127) at (11.85, 6.125) {n};
\\node [] (128) at (11.75, 4.875) {};
\\node [] (129) at (11.75, 2.875) {};
\\node [style=none, fill=white, right] (130) at (11.85, 4.775) {n};
\\node [] (131) at (10.5, 2.375) {};
\\node [] (132) at (10.5, 0.375) {};
\\node [style=none, fill=white, right] (133) at (10.6, 2.275) {s};
\\node [] (134) at (10.5, -0.125) {};
\\node [] (135) at (10.5, -0.875) {};
\\node [style=none, fill=white, right] (136) at (10.6, -0.225) {s};
\\node [] (137) at (13.0, 2.375) {};
\\node [] (138) at (13.0, -0.875) {};
\\node [style=none, fill=white, right] (139) at (13.1, 2.275) {s};
\\node [] (140) at (6.5, -1.875) {};
\\node [] (141) at (6.5, -3.875) {};
\\node [style=none, fill=white, right] (142) at (6.6, -1.975) {n};
\\node [] (143) at (9.0, -1.875) {};
\\node [] (144) at (9.0, -3.875) {};
\\node [style=none, fill=white, right] (145) at (9.1, -1.975) {n};
\\node [] (146) at (4.0, -1.875) {};
\\node [] (147) at (4.0, -3.875) {};
\\node [style=none, fill=white, right] (148) at (4.1, -1.975) {n};
\\node [] (149) at (9.0, -4.375) {};
\\node [] (150) at (9.0, -6.375) {};
\\node [style=none, fill=white, right] (151) at (9.1, -4.475) {n};
\\node [] (152) at (4.0, -4.375) {};
\\node [] (153) at (4.0, -11.875) {};
\\node [style=none, fill=white, right] (154) at (4.1, -4.475) {n};
\\node [] (155) at (6.5, -4.375) {};
\\node [] (156) at (6.5, -11.875) {};
\\node [style=none, fill=white, right] (157) at (6.6, -4.475) {n};
\\end{pgfonlayer}
\\begin{pgfonlayer}{edgelayer}
\\draw [-, fill={gray}] (1.center) to (2.center) to (3.center) to (4.center) to (1.center);
\\draw [-, fill={gray}] (6.center) to (7.center) to (8.center) to (9.center) to (6.center);
\\draw [-, fill={gray}] (11.center) to (12.center) to (13.center) to (14.center) to (11.center);
\\draw [-, fill={gray}] (16.center) to (17.center) to (18.center) to (19.center) to (16.center);
\\draw [-, fill={gray}] (21.center) to (22.center) to (23.center) to (24.center) to (21.center);
\\draw [-, fill={#fff9e5}] (26.center) to (27.center) to (28.center) to (29.center) to (26.center);
\\draw [-, fill={white}] (31.center) to (32.center) to (33.center) to (34.center) to (31.center);
\\draw [-, fill={gray}] (35.center) to (36.center) to (37.center) to (38.center) to (35.center);
\\draw [-, fill={white}] (40.center) to (41.center) to (42.center) to (43.center) to (40.center);
\\draw [-, fill={gray}] (44.center) to (45.center) to (46.center) to (47.center) to (44.center);
\\draw [-, fill={white}] (49.center) to (50.center) to (51.center) to (52.center) to (49.center);
\\draw [-, fill={gray}] (53.center) to (54.center) to (55.center) to (56.center) to (53.center);
\\draw [-, fill={gray}] (58.center) to (59.center) to (60.center) to (61.center) to (58.center);
\\draw [-, fill={gray}] (63.center) to (64.center) to (65.center) to (66.center) to (63.center);
\\draw [-, fill={gray}] (68.center) to (69.center) to (70.center) to (71.center) to (68.center);
\\draw [-, fill={gray}] (73.center) to (74.center) to (75.center) to (76.center) to (73.center);
\\draw [-, fill={gray}] (78.center) to (79.center) to (80.center) to (81.center) to (78.center);
\\draw [in=90, out=-90] (83.center) to (84.center);
\\draw [in=90, out=-90] (86.center) to (87.center);
\\draw [in=90, out=-90] (89.center) to (90.center);
\\draw [in=90, out=-90] (92.center) to (93.center);
\\draw [in=90, out=-90] (95.center) to (96.center);
\\draw [in=90, out=-90] (98.center) to (99.center);
\\draw [in=90, out=-90] (101.center) to (102.center);
\\draw [in=90, out=-90] (104.center) to (105.center);
\\draw [in=90, out=-90] (107.center) to (108.center);
\\draw [in=90, out=-90] (110.center) to (111.center);
\\draw [in=90, out=-90] (113.center) to (114.center);
\\draw [in=90, out=-90] (116.center) to (117.center);
\\draw [in=90, out=-90] (119.center) to (120.center);
\\draw [in=90, out=-90] (122.center) to (123.center);
\\draw [in=90, out=-90] (125.center) to (126.center);
\\draw [in=90, out=-90] (128.center) to (129.center);
\\draw [in=90, out=-90] (131.center) to (132.center);
\\draw [in=90, out=-90] (134.center) to (135.center);
\\draw [in=90, out=-90] (137.center) to (138.center);
\\draw [in=90, out=-90] (140.center) to (141.center);
\\draw [in=90, out=-90] (143.center) to (144.center);
\\draw [in=90, out=-90] (146.center) to (147.center);
\\draw [in=90, out=-90] (149.center) to (150.center);
\\draw [in=90, out=-90] (152.center) to (153.center);
\\draw [in=90, out=-90] (155.center) to (156.center);
\\end{pgfonlayer}
\\end{tikzpicture}
""",
]

@pytest.mark.parametrize('diagram, drawable', zip(diagrams, expected_drawables))
def test_drawable_generation(diagram, drawable):

Expand All @@ -398,3 +609,12 @@ def test_tikz_drawing(diagram, tikz, capsys):
tikz_op, _ = capsys.readouterr()

assert tikz_op == tikz


@pytest.mark.parametrize('diagram, tikz', zip(diagrams, colored_tikz_outputs))
def test_tikz_colored_drawing(diagram, tikz, capsys):

diagram.draw(backend=TikzBackend(), color_boxes=True)
tikz_op, _ = capsys.readouterr()

assert tikz_op == tikz

0 comments on commit 8af37b1

Please sign in to comment.