Add property "graph_type" to graphs

Author: prmtl

pydot_ng/__init__.py

@@ -1125,6 +1125,8 @@ def get_type(self):
         """Get the graph's type, 'graph' or 'digraph'."""
         return self.obj_dict['type']
 
+    graph_type = property(get_type, set_type)
+
     def set_name(self, graph_name):
         """Set the graph's name."""
         self.obj_dict['name'] = graph_name

test/test_pydot.py

@@ -39,6 +39,7 @@ def digraph():
 def test_keep_graph_type(graph_type):
     graph = pydot.Dot(graph_name="Test", graph_type=graph_type)
     assert graph.get_type() == graph_type
+    assert graph.graph_type == graph_type
 
 
 def test_node_style():