diff --git a/src/test/java/graph/EdgeTest.java b/src/test/java/graph/EdgeTest.java index d722e8d..67ccf41 100644 --- a/src/test/java/graph/EdgeTest.java +++ b/src/test/java/graph/EdgeTest.java @@ -16,16 +16,18 @@ void constructorWeight() @Test void constructorWeightAndDestination() { - Edge edge = new Edge(new Vertex("A"), 1.5f); + Vertex vertex = new Vertex("A"); + Edge edge = new Edge(vertex, 1.5f); assertEquals(1.5f, edge.getWeight()); - assertEquals(new Vertex("A"), edge.getDestination()); + assertEquals(vertex, edge.getDestination()); } @Test void getDestination() { - Edge edge = new Edge(new Vertex("A"), 1.5f); - assertEquals(new Vertex("A"), edge.getDestination()); + Vertex vertex = new Vertex("A"); + Edge edge = new Edge(vertex, 1.5f); + assertEquals(vertex, edge.getDestination()); } @Test @@ -34,7 +36,7 @@ void setDestination() Vertex vertex = new Vertex("A"); Edge edge = new Edge(1.5f); edge.setDestination(vertex); - assertEquals(new Vertex("A"), edge.getDestination()); + assertEquals(vertex, edge.getDestination()); } @Test