Add tests

Author: curran

.gitignore

@@ -1,2 +1,3 @@
 node_modules
 package-lock.json
+*.swp

index.js

@@ -88,6 +88,7 @@ module.exports = function Graph(serialized){
   // Sets the weight of the given edge.
   function setEdgeWeight(u, v, weight){
     edgeWeights[encodeEdge(u, v)] = weight;
+    return graph;
   }
 
   // Gets the weight of the given edge.

test.js

@@ -313,10 +313,24 @@ describe("Graph", function() {
   });
 
   describe("Edge Weights", function() {
+
     it("Should set and get an edge weight.", function (){
       var graph = Graph().addEdge("a", "b", 5);
       assert.equal(graph.getEdgeWeight("a", "b"), 5);
     });
+
+    it("Should set edge weight via setEdgeWeight.", function (){
+      var graph = Graph()
+        .addEdge("a", "b")
+        .setEdgeWeight("a", "b", 5);
+      assert.equal(graph.getEdgeWeight("a", "b"), 5);
+    });
+
+    it("Should return weight of 1 if no weight set.", function (){
+      var graph = Graph().addEdge("a", "b");
+      assert.equal(graph.getEdgeWeight("a", "b"), 1);
+    });
+
   });
 });