[DO NOT MERGE] add_pymmmk_istvan

lowkey/MMMKModel.py

@@ -2,14 +2,29 @@
 #  Copyright 2011 by the AToMPM team and licensed under the LGPL
 #  See COPYING.lesser and README.md in the root of this project for full details
 
-from lowkey.lww.LWWGraph import LWWGraph
+from lowkey.lww.LWWMap import LWWMap
 
-__author__ = "Istvan David, Bentley James Oakes"
+__author__ = "Istvan David"
 __copyright__ = "Copyright 2022, GEODES"
 __credits__ = "Eugene Syriani"
 __license__ = "GPL-3.0"
 
-class Model(LWWGraph):
-    def __init__(self):
+class Model(LWWMap):
+    def __init__(self, from_dict : dict = None):
         super().__init__()
-        self.metamodels = []
+
+        self.nodes = {}
+        self.edges = []
+        self.metamodels = []
+
+        if from_dict:
+            self.nodes = from_dict["nodes"]
+            self.edges = from_dict["edges"]
+            self.metamodels = from_dict["metamodels"]
+
+    def to_dict(self) -> dict:
+        return {
+            "nodes": self.nodes,
+            "edges": self.edges,
+            "metamodels": self.metamodels,
+        }

lowkey/MMMKModelEdge.py

@@ -4,7 +4,7 @@
 
 from lowkey.lww.LWWEdge import LWWEdge
 
-__author__ = "Istvan David, Bentley James Oakes"
+__author__ = "Istvan David"
 __copyright__ = "Copyright 2022, GEODES"
 __credits__ = "Eugene Syriani"
 __license__ = "GPL-3.0"

lowkey/MMMKModelInterfaceTests.py

@@ -0,0 +1,49 @@
+#!/usr/bin/env python
+import unittest
+
+from MMMKModel import Model
+from MMMKModelNode import Node
+from MMMKModelEdge import Edge
+
+__author__ = "Istvan David"
+__copyright__ = "Copyright 2022, GEODES"
+__credits__ = "Eugene Syriani"
+__license__ = "GPL-3.0"
+
+
+class MMMKModelLWWTests(unittest.TestCase):
+
+    model = None
+
+    def setUp(self):
+        self.model = Model()
+
+    def tearDown(self):
+        self.model = None
+
+    def testAddVertices(self):
+        v1Name = "A"
+        v2Name = "B"
+
+        v1 = Node()
+        v1.setName(v1Name)
+        v2 = Node()
+        v2.setName(v2Name)
+
+        self.model.addVertex(v1, 10)
+        self.assertTrue(self.model.vertexExists(v1))
+        self.assertEqual(self.model.numberOfVertices(), 1)
+        v1AdjacencySet = self.model.getAdjacencyListForVertex(v1)
+        self.assertEqual(len(v1AdjacencySet), 0)
+
+        self.model.addVertex(v2, 30)
+        self.assertTrue(self.model.vertexExists(v1))
+        self.assertTrue(self.model.vertexExists(v2))
+        self.assertEqual(self.model.numberOfVertices(), 2)
+        v1AdjacencySet = self.model.getAdjacencyListForVertex(v1)
+        v2AdjacencySet = self.model.getAdjacencyListForVertex(v2)
+        self.assertEqual(len(v1AdjacencySet), 0)
+        self.assertEqual(len(v2AdjacencySet), 0)
+
+if __name__ == "__main__":
+    unittest.main()