Add getNeighbors method to Graph.

trekhleb · trekhleb · commit ddd7f9fe0d8f · 2018-04-11T08:42:17.000+03:00
diff --git a/src/data-structures/graph/Graph.js b/src/data-structures/graph/Graph.js
@@ -25,6 +25,13 @@ export default class Graph {
     return this.vertices[vertexKey];
   }
 
+  /**
+   * @param {GraphVertex} vertex
+   */
+  getNeighbors(vertex) {
+    return vertex.getNeighbors();
+  }
+
   /**
    * @param {GraphEdge} edge
    * @returns {Graph}
diff --git a/src/data-structures/graph/__test__/Graph.test.js b/src/data-structures/graph/__test__/Graph.test.js
@@ -111,4 +111,25 @@ describe('Graph', () => {
     expect(graphEdgeAB).toEqual(edgeAB);
     expect(graphEdgeAB.weight).toBe(10);
   });
+
+  it('should return vertex neighbors', () => {
+    const graph = new Graph(true);
+
+    const vertexA = new GraphVertex('A');
+    const vertexB = new GraphVertex('B');
+    const vertexC = new GraphVertex('C');
+
+    const edgeAB = new GraphEdge(vertexA, vertexB);
+    const edgeAC = new GraphEdge(vertexA, vertexC);
+
+    graph
+      .addEdge(edgeAB)
+      .addEdge(edgeAC);
+
+    const neighbors = graph.getNeighbors(vertexA);
+
+    expect(neighbors.length).toBe(2);
+    expect(neighbors[0]).toEqual(vertexB);
+    expect(neighbors[1]).toEqual(vertexC);
+  });
 });
