Merge pull request #81 from clue/fix-mbf

Fix Moore-Bellman-Ford for unweighted edges

Author: clue

src/ShortestPath/MooreBellmanFord.php

@@ -43,7 +43,7 @@ private function bigStep(Edges $edges, array &$totalCostOfCheapestPathTo, array
                     // New possible costs of this path
                     $newCost = $totalCostOfCheapestPathTo[$fromVertex->getId()] + $edge->getWeight();
                     if (is_infinite($newCost)) {
-                        $newCost = $edge->getWeight();
+                        $newCost = $edge->getWeight() + 0;
                     }
 
                     // No path has been found yet

tests/ShortestPath/BaseShortestPathTest.php

@@ -116,6 +116,24 @@ public function testSeparateGraphsAreNotReachable()
         $alg->getEdgesTo($vg2);
     }
 
+    public function testGraphUnweighted()
+    {
+        // 1 -> 2
+        $graph = new Graph();
+        $v1 = $graph->createVertex(1);
+        $v2 = $graph->createVertex(2);
+        $e1 = $v1->createEdgeTo($v2);
+
+        $alg = $this->createAlg($v1);
+
+        $expectedWeight = $this->getExpectedWeight(array($e1));
+        $this->assertEquals($expectedWeight, $alg->getDistance($v2));
+        $this->assertEquals(array(2 => $expectedWeight), $alg->getDistanceMap());
+        $this->assertEquals(array($e1), $alg->getEdges()->getVector());
+        $this->assertEquals(array($e1), $alg->getEdgesTo($v2)->getVector());
+        $this->assertEquals(array(2), $alg->getVertices()->getIds());
+    }
+
     public function testGraphTwoComponents()
     {
         // 1 -[10]-> 2

tests/ShortestPath/MooreBellmanFordTest.php

@@ -69,4 +69,28 @@ public function testLoopNegativeWeightIsCycle()
 
         $cycle = $alg->getCycleNegative();
     }
+
+    public function testNegativeComponentHasCycle()
+    {
+        // 1 -[1]-> 2     3 --[-1]--> 4
+        //                ^           |
+        //                \---[-2]----/
+        $graph = new Graph();
+        $v1 = $graph->createVertex(1);
+        $v2 = $graph->createVertex(2);
+        $v3 = $graph->createVertex(3);
+        $v4 = $graph->createVertex(4);
+        $e1 = $v1->createEdgeTo($v2)->setWeight(1);
+        $e2 = $v3->createEdgeTo($v4)->setWeight(-1);
+        $e3 = $v4->createEdgeTo($v3)->setWeight(-2);
+
+        // second component has a cycle
+        $alg = $this->createAlg($v3);
+        $cycle = $alg->getCycleNegative();
+
+        // first component does not have a cycle
+        $alg = $this->createAlg($v1);
+        $this->setExpectedException('UnderflowException');
+        $alg->getCycleNegative();
+    }
 }