fix(diagram): normalize case-insensitive Graphviz attributes

pull/1260/head
Your Name 4 days ago
parent 0cd0d84016
commit a2a2c37f9a

@ -196,11 +196,11 @@ class Diagram:
if not self._validate_direction(direction):
raise ValueError(f'"{direction}" is not a valid direction')
self.dot.graph_attr["rankdir"] = direction
self.dot.graph_attr["rankdir"] = direction.upper()
if not self._validate_curvestyle(curvestyle):
raise ValueError(f'"{curvestyle}" is not a valid curvestyle')
self.dot.graph_attr["splines"] = curvestyle
self.dot.graph_attr["splines"] = curvestyle.lower()
if isinstance(outformat, list):
for one_format in outformat:

@ -43,6 +43,11 @@ class DiagramTest(unittest.TestCase):
with self.assertRaises(ValueError):
Diagram(curvestyle=cvs)
def test_case_insensitive_graph_attributes(self):
diagram = Diagram(direction="lr", curvestyle="CURVED")
self.assertEqual(diagram.dot.graph_attr["rankdir"], "LR")
self.assertEqual(diagram.dot.graph_attr["splines"], "curved")
def test_validate_theme(self):
# Valid themes.
for theme in ("neutral", "pastel", "blues", "greens", "orange"):

Loading…
Cancel
Save