test: add tests for curvestyle and theme validation

pull/1200/head
mingrammer 1 month ago
parent ff80fc5c0c
commit 520137a2a8

@ -34,15 +34,25 @@ class DiagramTest(unittest.TestCase):
Diagram(direction=dir)
def test_validate_curvestyle(self):
# Normal directions.
for cvs in ("ortho", "curved", "CURVED"):
# Normal curvestyles.
for cvs in ("ortho", "curved", "spline", "polyline", "CURVED"):
Diagram(curvestyle=cvs)
# Invalid directions.
# Invalid curvestyles.
for cvs in ("tangent", "unknown"):
with self.assertRaises(ValueError):
Diagram(curvestyle=cvs)
def test_validate_theme(self):
# Valid themes.
for theme in ("neutral", "pastel", "blues", "greens", "orange"):
Diagram(theme=theme)
# Invalid themes.
for theme in ("dark", "unknown"):
with self.assertRaises(ValueError):
Diagram(theme=theme)
def test_validate_outformat(self):
# Normal output formats.
for fmt in ("png", "jpg", "svg", "pdf", "PNG", "dot"):

Loading…
Cancel
Save