Fix game tests for random dificulty

pedrohml · Dec 24, 2015 · c424b1c · c424b1c
1 parent d878b7a
commit c424b1c
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/lib/game/game.rb b/lib/game/game.rb
@@ -17,13 +17,13 @@ def select_goal_cell(dificulty)
 		depths = leafs.keys.sort
 		case dificulty
 		when :easy
-			selected_depth = depths[(depths.size*0.25).ceil]
+			selected_depth = depths[(depths.size/4).ceil]
 		when :medium
-			selected_depth = depths[(depths.size*0.5).ceil]
+			selected_depth = depths[(depths.size/2).ceil]
 		when :hard
 			selected_depth = depths.last
 		else
-			selected_depth = depths[rand((depths.size/2.0).ceil) + (depths.size/2.0).ceil]
+			selected_depth = depths[(depths.size/2.0).floor + rand((depths.size/2.0).round)]
 		end
 		leafs[selected_depth].content
 	end

diff --git a/spec/game/game_spec.rb b/spec/game/game_spec.rb
@@ -16,17 +16,17 @@
 		expect(maze.matrix.flatten).to include(2)
 	end
 
-	it '#generate_goal :easy' do
+	it '#generate_goal :medium' do
 		maze = @game.generate_goal(:medium)
 		expect(maze.matrix.flatten).to include(2)
 	end
 
-	it '#generate_goal :easy' do
+	it '#generate_goal :hard' do
 		maze = @game.generate_goal(:hard)
 		expect(maze.matrix.flatten).to include(2)
 	end
 
-	it '#generate_goal :easy' do
+	it '#generate_goal :random' do
 		maze = @game.generate_goal(:random)
 		expect(maze.matrix.flatten).to include(2)
 	end