feat(constructor): Add nodeIDs set (#34)

Author: Sirivasv

src/bmssp.mjs

@@ -1,9 +1,16 @@
 class BMSSP {
   constructor(inputGraph) {
+    // Main graph represented as an array of edges
     this.graph = [];
+    // Set to store unique node IDs
+    this.nodeIDs = new Set();
+
     for (let edge of inputGraph) {
       // Create a deep copy of each edge array
       this.graph.push([...edge]);
+      // Add node IDs to the set
+      this.nodeIDs.add(edge[0]);
+      this.nodeIDs.add(edge[1]);
     }
   }
 }

test/main.test.mjs

@@ -25,3 +25,15 @@ describe("BMSSP constructor", () => {
     expect(myBMSSP.graph).toEqual(roadNetCA);
   });
 });
+
+describe("BMSSP nodeIDs", () => {
+  test("stores unique node IDs correctly", () => {
+    const myBMSSP = new BMSSP(roadNetCA);
+    const uniqueNodeIDs = new Set();
+    roadNetCA.forEach((edge) => {
+      uniqueNodeIDs.add(edge[0]);
+      uniqueNodeIDs.add(edge[1]);
+    });
+    expect(myBMSSP.nodeIDs).toEqual(uniqueNodeIDs);
+  });
+});