Fixed unit tests.

wingedsheep · May 19, 2020 · d100aaf · d100aaf
1 parent b0c8bd1
commit d100aaf
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 6 deletions.
diff --git a/test/city_util_test.py b/test/city_util_test.py
@@ -1,8 +1,6 @@
 import unittest
 
 from main.carcassonne_game_state import CarcassonneGameState
-from main.carcassonne_visualiser import CarcassonneVisualiser
-from main.tile_sets import inns_and_cathedrals_deck
 from main.tile_sets.base_deck import base_tiles
 from main.tile_sets.inns_and_cathedrals_deck import inns_and_cathedrals_tiles
 from main.objects.city import City
@@ -24,7 +22,7 @@ def test_find_city(self):
         # Given
         game_state: CarcassonneGameState = CarcassonneGameState()
 
-        city_top = base_tiles["city_one_side"]
+        city_top = base_tiles["city_top"]
         city_bottom = city_top.turn(2)
 
         game_state.board = [[None for column in range(1)] for row in range(2)]
@@ -138,7 +136,7 @@ def test_find_cities(self):
         # Given
         game_state: CarcassonneGameState = CarcassonneGameState()
 
-        city_one_side_straight_road = base_tiles["city_one_side_straight_road"].turn(3)
+        city_one_side_straight_road = base_tiles["city_top_straight_road"].turn(3)
         city_with_road = inns_and_cathedrals_tiles["ic_15"].turn(3)
 
         game_state.board = [[None for column in range(2)] for row in range(1)]

diff --git a/test/points_collector_test.py b/test/points_collector_test.py
@@ -21,7 +21,7 @@ def test_collect_points_small_city(self):
         # Given
         game_state: CarcassonneGameState = CarcassonneGameState()
 
-        city_one_side_straight_road = base_tiles["city_one_side_straight_road"].turn(3)
+        city_one_side_straight_road = base_tiles["city_top_straight_road"].turn(3)
         city_with_road = inns_and_cathedrals_tiles["ic_15"].turn(3)
 
         game_state.board = [[None for column in range(2)] for row in range(1)]
@@ -52,7 +52,7 @@ def test_collect_points_small_city_2(self):
         # Given
         game_state: CarcassonneGameState = CarcassonneGameState()
 
-        city_one_side_straight_road = base_tiles["city_one_side_straight_road"].turn(3)
+        city_one_side_straight_road = base_tiles["city_top_straight_road"].turn(3)
         city_with_road = inns_and_cathedrals_tiles["ic_15"].turn(3)
 
         game_state.board = [[None for column in range(2)] for row in range(1)]