Update tests to new exceptions

fizzlebert · Dec 20, 2018 · c2db53d · c2db53d
1 parent 717c78b
commit c2db53d
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 5 deletions.
diff --git a/tests/context.py b/tests/context.py
@@ -5,4 +5,5 @@
 
 from amazing.dijkstra import Dijkstra
 from amazing.astar import Astar
-from amazing import errors
+from amazing.maze import Maze
+from amazing import exceptions, maze
diff --git a/tests/test_astar.py b/tests/test_astar.py
@@ -1,4 +1,4 @@
-from context import Astar, errors
+from context import Astar, exceptions
 
 import pytest
 
@@ -17,5 +17,5 @@ def test_scenario():
 
 
 def test_unknown_node():
-    with pytest.raises(errors.NodeNotFound):
+    with pytest.raises(exceptions.NodeNotFound):
         graph.shortest_path("a", "g")
diff --git a/tests/test_dijkstra.py b/tests/test_dijkstra.py
@@ -1,4 +1,4 @@
-from context import Dijkstra, errors
+from context import Dijkstra, exceptions
 
 import pytest
 
@@ -17,5 +17,5 @@ def test_scenario():
 
 
 def test_unknown_node():
-    with pytest.raises(errors.NodeNotFound):
+    with pytest.raises(exceptions.NodeNotFound):
         graph.shortest_path("a", "g")