$:.unshift File.join(File.dirname(__FILE__), "..")
require 'test/unit'
require 'shogi_server'
+require 'shogi_server/league.rb'
require 'shogi_server/player'
require 'shogi_server/pairing'
require 'test/mock_log_message'
-
def same_pair?(a, b)
unless a.size == 2 && b.size == 2
return false
end
end
+class TestLeastDiff < Test::Unit::TestCase
+
+ class MockLeague
+ def initialize
+ @players = []
+ end
+
+ def add(player)
+ @players << player
+ end
+
+ def find(name)
+ @players.find do |p|
+ p.name == name
+ end
+ end
+ end
+
+ def setup
+ $league = MockLeague.new
+
+ @pairing= ShogiServer::LeastDiff.new
+ $paired = []
+ $called = 0
+ def @pairing.start_game(p1,p2)
+ $called += 1
+ $paired << [p1,p2]
+ end
+
+ @file = Pathname.new(File.join(File.dirname(__FILE__), "floodgate_history.yaml"))
+ @history = ShogiServer::League::Floodgate::History.new @file
+
+ @a = ShogiServer::BasicPlayer.new
+ @a.player_id = "a"
+ @a.name = "a"
+ @a.win = 1
+ @a.loss = 2
+ @a.rate = 500
+ @b = ShogiServer::BasicPlayer.new
+ @b.player_id = "b"
+ @b.name = "b"
+ @b.win = 10
+ @b.loss = 20
+ @b.rate = 800
+ @c = ShogiServer::BasicPlayer.new
+ @c.player_id = "c"
+ @c.name = "c"
+ @c.win = 100
+ @c.loss = 200
+ @c.rate = 1000
+ @d = ShogiServer::BasicPlayer.new
+ @d.player_id = "d"
+ @d.name = "d"
+ @d.win = 1000
+ @d.loss = 2000
+ @d.rate = 1500
+ @e = ShogiServer::BasicPlayer.new
+ @e.player_id = "e"
+ @e.name = "e"
+ @e.win = 3000
+ @e.loss = 3000
+ @e.rate = 2000
+ @f = ShogiServer::BasicPlayer.new
+ @f.player_id = "f"
+ @f.name = "f"
+ @f.win = 4000
+ @f.loss = 4000
+ @f.rate = 2150
+ @g = ShogiServer::BasicPlayer.new
+ @g.player_id = "g"
+ @g.name = "g"
+ @g.win = 5000
+ @g.loss = 5000
+ @g.rate = 2500
+ @h = ShogiServer::BasicPlayer.new
+ @h.player_id = "h"
+ @h.name = "h"
+ @h.win = 6000
+ @h.loss = 6000
+ @h.rate = 3000
+ @x = ShogiServer::BasicPlayer.new
+ @x.player_id = "x"
+ @x.name = "x"
+
+ @abcdefg1 = ShogiServer::BasicPlayer.new
+ @abcdefg1.player_id = "abcdefg1"
+ @abcdefg1.name = "abcdefg1"
+ @abcdefg1.rate = 2100
+ @abcdefg2 = ShogiServer::BasicPlayer.new
+ @abcdefg2.player_id = "abcdefg2"
+ @abcdefg2.name = "abcdefg2"
+ @abcdefg2.rate = 2200
+ @abcdxyz = ShogiServer::BasicPlayer.new
+ @abcdxyz.player_id = "abcdxyz"
+ @abcdxyz.name = "abcdxyz"
+ @abcdxyz.rate = 2300
+
+ $league.add(@a)
+ $league.add(@b)
+ $league.add(@c)
+ $league.add(@d)
+ $league.add(@e)
+ $league.add(@f)
+ $league.add(@g)
+ $league.add(@h)
+ $league.add(@x)
+ $league.add(@abcdefg1)
+ $league.add(@abcdefg2)
+ $league.add(@abcdxyz)
+ end
+
+ def teardown
+ @file.delete if @file.exist?
+ end
+
+ def assert_pairs(x_array, y_array)
+ if (x_array.size != y_array.size)
+ assert_equal(x_array.size, y_array.size)
+ return
+ end
+ i = 0
+
+ if (x_array.size == 1)
+ assert_equal(x_array[0].name, y_array[0].name)
+ return
+ end
+
+ ret = true
+ while i < x_array.size
+ if i == x_array.size-1
+ assert_equal(x_array[i].name, y_array[i].name)
+ break
+ end
+ px1 = x_array[i]
+ px2 = x_array[i+1]
+ py1 = y_array[i]
+ py2 = y_array[i+1]
+
+ if ! ((px1.name == py1.name && px2.name == py2.name) ||
+ (px1.name == py2.name && px2.name == py1.name))
+ ret = false
+ end
+ i += 2
+ end
+
+ assert(ret)
+ end
+
+ def test_match_one_player
+ players = [@a]
+ assert_equal(0, @pairing.calculate_diff_with_penalty(players,nil))
+ r = @pairing.match(players)
+ assert_pairs([@a], r)
+ end
+
+ def test_match_two_players
+ players = [@a,@b]
+ assert_equal(@b.rate-@a.rate, @pairing.calculate_diff_with_penalty([@a,@b],nil))
+ assert_equal(@b.rate-@a.rate, @pairing.calculate_diff_with_penalty([@b,@a],nil))
+ r = @pairing.match(players)
+ assert_pairs([@a,@b], r)
+ end
+
+ def test_match_three_players
+ players = [@h,@a,@b]
+ assert_equal(300, @pairing.calculate_diff_with_penalty([@a,@b,@h],nil))
+ assert_equal(2200, @pairing.calculate_diff_with_penalty([@b,@h,@a],nil))
+ r = @pairing.match(players)
+ assert_pairs([@a,@b,@h], r)
+ assert_pairs([@a,@b,@h], players)
+ end
+
+ def test_match_many_players
+ players = [@a,@b,@h,@a,@b,@h,@a,@b,@h,@a,@b,@h,@a,@b,@h,@a,@b,@h,@a,@b,@h,@a,@b,@h,@a,@b,@h,@a,@b,@h]
+ r = @pairing.match(players)
+ assert true
+ end
+
+ def test_calculate_diff_with_penalty
+ players = [@a,@b]
+ assert_equal(@b.rate-@a.rate, @pairing.calculate_diff_with_penalty(players,nil))
+
+ dummy = nil
+ def @history.make_record(game_result)
+ {:game_id => "wdoor+floodgate-900-0-a-b-1",
+ :black => "b", :white => "a",
+ :winner => "a", :loser => "b"}
+ end
+ @history.update(dummy)
+ assert_equal(@b.rate-@a.rate+400, @pairing.calculate_diff_with_penalty(players, @history))
+ end
+
+ def test_calculate_diff_with_penalty2
+ players = [@a,@b,@g,@h]
+ assert_equal(@b.rate-@a.rate+@h.rate-@g.rate, @pairing.calculate_diff_with_penalty(players,nil))
+ end
+
+ def test_calculate_diff_with_penalty2_1
+ players = [@a,@b,@g,@h]
+ assert_equal(@b.rate-@a.rate+@h.rate-@g.rate, @pairing.calculate_diff_with_penalty(players,nil))
+ dummy = nil
+ def @history.make_record(game_result)
+ {:game_id => "wdoor+floodgate-900-0-a-b-1",
+ :black => "b", :white => "a",
+ :winner => "a", :loser => "b"}
+ end
+ @history.update(dummy)
+ assert_equal(@b.rate-@a.rate+400+@h.rate-@g.rate, @pairing.calculate_diff_with_penalty(players, @history))
+ end
+
+ def test_calculate_diff_with_penalty2_2
+ players = [@a,@b,@g,@h]
+ assert_equal(@b.rate-@a.rate+@h.rate-@g.rate, @pairing.calculate_diff_with_penalty(players,nil))
+ dummy = nil
+ def @history.make_record(game_result)
+ {:game_id => "wdoor+floodgate-900-0-a-b-1",
+ :black => "g", :white => "h",
+ :winner => "h", :loser => "g"}
+ end
+ @history.update(dummy)
+ assert_equal(@b.rate-@a.rate+400+@h.rate-@g.rate, @pairing.calculate_diff_with_penalty(players, @history))
+ #assert_equal(@b.rate-@a.rate+400+@h.rate-@g.rate+400, @pairing.calculate_diff_with_penalty(players, [@b,@a,@h,@g]))
+ end
+
+ def test_calculate_diff_with_penalty2_3
+ players = [@a,@b,@g,@h]
+ assert_equal(@b.rate-@a.rate+@h.rate-@g.rate, @pairing.calculate_diff_with_penalty(players,nil))
+ dummy = nil
+ def @history.make_record(game_result)
+ {:game_id => "wdoor+floodgate-900-0-a-b-1",
+ :black => "g", :white => "h",
+ :winner => "h", :loser => "g"}
+ end
+ @history.update(dummy)
+ def @history.make_record(game_result)
+ {:game_id => "wdoor+floodgate-900-0-a-b-1",
+ :black => "b", :white => "a",
+ :winner => "a", :loser => "b"}
+ end
+ @history.update(dummy)
+ assert_equal(@b.rate-@a.rate+400+@h.rate-@g.rate+400, @pairing.calculate_diff_with_penalty(players, @history))
+ end
+
+ def test_calculate_diff_with_kin_4_players
+ players = [@abcdefg1, @abcdxyz]
+ assert_equal(@abcdxyz.rate - @abcdefg1.rate + 400, @pairing.calculate_diff_with_penalty(players,nil))
+ end
+
+ def test_calculate_diff_with_kin_7_players
+ players = [@abcdefg1, @abcdefg2]
+ assert_equal(@abcdefg2.rate - @abcdefg1.rate + 800, @pairing.calculate_diff_with_penalty(players,nil))
+ end
+
+ def test_get_player_rate_0
+ assert_equal(2150, @pairing.get_player_rate(@x, @history))
+
+ @x.estimated_rate = 0
+ dummy = nil
+ def @history.make_record(game_result)
+ {:game_id => "wdoor+floodgate-900-0-x-a-1",
+ :black => "x", :white => "a",
+ :winner => "x", :loser => "a"}
+ end
+ @history.update(dummy)
+ assert_equal(@a.rate+200, @pairing.get_player_rate(@x, @history))
+
+ @x.estimated_rate = 0
+ def @history.make_record(game_result)
+ {:game_id => "wdoor+floodgate-900-0-x-b-1",
+ :black => "x", :white => "b",
+ :winner => "b", :loser => "x"}
+ end
+ @history.update(dummy)
+
+ assert_equal(@b.rate-200, @pairing.get_player_rate(@x, @history))
+ end
+
+ def test_total_posibilities
+ assert_equal 1, @pairing.total_posibilities(2)
+ assert_equal 1, @pairing.total_posibilities(3)
+ assert_equal 3, @pairing.total_posibilities(4)
+ assert_equal 945, @pairing.total_posibilities(10)
+ end
+end
+
+class TestExcludeUnratedPlayers < Test::Unit::TestCase
+ def setup
+ @pairing= ShogiServer::ExcludeUnratedPlayers.new
+ @a = ShogiServer::BasicPlayer.new
+ @a.name = "a"
+ @a.win = 1
+ @a.loss = 2
+ @a.rate = 0
+ @b = ShogiServer::BasicPlayer.new
+ @b.name = "b"
+ @b.win = 10
+ @b.loss = 20
+ @b.rate = 1500
+ @c = ShogiServer::BasicPlayer.new
+ @c.name = "c"
+ @c.win = 100
+ @c.loss = 200
+ @c.rate = 1000
+ @d = ShogiServer::BasicPlayer.new
+ @d.name = "d"
+ @d.win = 1000
+ @d.loss = 2000
+ @d.rate = 2000
+ end
+
+ def test_match_without_any_players
+ players = []
+ @pairing.match(players)
+ assert_equal([], players)
+ end
+
+ def test_match_without_unrated_player_1
+ players = [@b, @c, @d]
+ @pairing.match(players)
+ assert_equal([@b, @c, @d], players)
+ end
+
+ def test_match_without_unrated_player_2
+ players = [@b]
+ @pairing.match(players)
+ assert_equal([@b], players)
+ end
+
+ def test_match_with_unrated_player_1
+ players = [@a, @b, @c, @d]
+ @pairing.match(players)
+ assert_equal([@b, @c, @d], players)
+ end
+ def test_match_with_unrated_player_2
+ players = [@a]
+ @pairing.match(players)
+ assert_equal([], players)
+ end
+end
OSDN Git Service
