Merge pull request #6 from rohaquinlop/dev

feat(view): #4 enable graphviz customization
rohaquinlop · Nov 5, 2023 · 4e2f798 · 4e2f798
2 parents d8f7558 + 5c1ebd3
commit 4e2f798
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 4 deletions.
diff --git a/automathon/finiteAutomata/dfa.py b/automathon/finiteAutomata/dfa.py
@@ -1,4 +1,6 @@
 # Exceptions module
+from __future__ import annotations
+
 from automathon.errors.errors import *
 from collections import deque
 from graphviz import Digraph
@@ -218,8 +220,8 @@ def union(self, M: 'DFA') -> 'DFA':
 
         return tmp_nfa.get_dfa()
 
-    def view(self, fileName: str):
-        dot = Digraph(name=fileName, format='png')
+    def view(self, file_name: str, node_attr: dict[str, str] | None = None, edge_attr: dict[str, str] | None = None):
+        dot = Digraph(name=file_name, format='png', node_attr=node_attr, edge_attr=edge_attr)
 
         dot.graph_attr['rankdir'] = 'LR'
 

diff --git a/automathon/finiteAutomata/nfa.py b/automathon/finiteAutomata/nfa.py
@@ -1,4 +1,6 @@
 # Exceptions module
+from __future__ import annotations
+
 from automathon.errors.errors import *
 from collections import deque
 from graphviz import Digraph
@@ -400,8 +402,8 @@ def product(self, M: 'NFA') -> 'NFA':
 
         return nfa
 
-    def view(self, file_name: str):
-        dot = Digraph(name=file_name, format='png')
+    def view(self, file_name: str, node_attr: dict[str, str] | None = None, edge_attr: dict[str, str] | None = None):
+        dot = Digraph(name=file_name, format='png', node_attr=node_attr, edge_attr=edge_attr)
 
         dot.graph_attr['rankdir'] = 'LR'
 

diff --git a/tests/test_nfa.py b/tests/test_nfa.py
@@ -181,3 +181,7 @@ def test_renumber(self):
         automata_4.renumber()
 
         self.assertTrue(automata_4.is_valid())
+
+
+if __name__ == '__main__':
+    unittest.main()