added testing

1 files changed, 8 insertions, 1 deletions

diff --git a/tikzit/src/data/tikzgraphassembler.cpp b/tikzit/src/data/tikzgraphassembler.cpp
index a785d85..d8e3963 100644
--- a/tikzit/src/data/tikzgraphassembler.cpp
+++ b/tikzit/src/data/tikzgraphassembler.cpp
@@ -1,9 +1,16 @@
 #include "tikzgraphassembler.h"
 
-TikzGraphAssembler::TikzGraphAssembler(QObject *parent) : QObject(parent)
+TikzGraphAssembler::TikzGraphAssembler(Graph *graph, QObject *parent) :
+    _graph(graph), QObject(parent)
 {
 
 }
 
 void TikzGraphAssembler::addNodeToMap(Node *n) { _nodeMap.insert(n->name(), n); }
 Node *TikzGraphAssembler::nodeWithName(QString name) { return _nodeMap[name]; }
+
+Graph *TikzGraphAssembler::graph() const
+{
+    return _graph;
+}
+