test/TC_pairing.rb

   1 $:.unshift File.join(File.dirname(__FILE__), "..")
   2 require 'test/unit'
   3 require 'shogi_server'
   4 require 'shogi_server/league.rb'
   5 require 'shogi_server/player'
   6 require 'shogi_server/pairing'
   7 require 'test/mock_log_message'
   8
   9 def same_pair?(a, b)
  10   unless a.size == 2 && b.size == 2
  11     return false
  12   end
  13
  14   return true if [a.first, a.last] == b || [a.last, a.first] == b
  15 end
  16
  17 class TestPairing < Test::Unit::TestCase
  18   def setup
  19     @pairing= ShogiServer::Pairing.new
  20     $pairs = []
  21     def @pairing.start_game(p1,p2)
  22       $pairs << [p1,p2]
  23     end
  24     @a = ShogiServer::BasicPlayer.new
  25     @a.name = "a"
  26     @a.win  = 1
  27     @a.loss = 2
  28     @a.rate = 0
  29     @a.last_game_win = false
  30     @b = ShogiServer::BasicPlayer.new
  31     @b.name = "b"
  32     @b.win  = 10
  33     @b.loss = 20
  34     @b.rate = 1500
  35     @b.last_game_win = true
  36     @c = ShogiServer::BasicPlayer.new
  37     @c.name = "c"
  38     @c.win  = 100
  39     @c.loss = 200
  40     @c.rate = 1000
  41     @c.last_game_win = true
  42     @d = ShogiServer::BasicPlayer.new
  43     @d.name = "d"
  44     @d.win  = 1000
  45     @d.loss = 2000
  46     @d.rate = 1800
  47     @d.last_game_win = true
  48   end
  49
  50   def test_include_newbie
  51     assert(@pairing.include_newbie?([@a]))
  52     assert(!@pairing.include_newbie?([@b]))
  53     assert(@pairing.include_newbie?([@b,@a]))
  54     assert(!@pairing.include_newbie?([@b,@c]))
  55   end
  56 end
  57
  58 class TestStartGame < Test::Unit::TestCase
  59   def setup
  60     @pairing= ShogiServer::StartGame.new
  61     $called = 0
  62     def @pairing.start_game(p1,p2)
  63       $called += 1
  64     end
  65     @a = ShogiServer::BasicPlayer.new
  66     @a.name = "a"
  67     @a.win  = 1
  68     @a.loss = 2
  69     @a.rate = 0
  70     @b = ShogiServer::BasicPlayer.new
  71     @b.name = "b"
  72     @b.win  = 10
  73     @b.loss = 20
  74     @b.rate = 1500
  75     @c = ShogiServer::BasicPlayer.new
  76     @c.name = "c"
  77     @c.win  = 100
  78     @c.loss = 200
  79     @c.rate = 1000
  80     @d = ShogiServer::BasicPlayer.new
  81     @d.name = "d"
  82     @d.win  = 1000
  83     @d.loss = 2000
  84     @d.rate = 2000
  85   end
  86
  87   def test_match_two_players
  88     players = [@a,@b]
  89     @pairing.match(players)
  90     assert_equal(1, $called)
  91   end
  92
  93   def test_match_one_player
  94     players = [@a]
  95     @pairing.match(players)
  96     assert_equal(0, $called)
  97   end
  98
  99   def test_match_zero_player
 100     players = []
 101     @pairing.match(players)
 102     assert_equal(0, $called)
 103   end
 104
 105   def test_match_three_players
 106     players = [@a,@b,@c]
 107     @pairing.match(players)
 108     assert_equal(1, $called)
 109   end
 110
 111   def test_match_four_players
 112     players = [@a,@b,@c,@d]
 113     @pairing.match(players)
 114     assert_equal(2, $called)
 115   end
 116 end
 117
 118 class TestStartGameWithoutHumans < Test::Unit::TestCase
 119   def setup
 120     @pairing= ShogiServer::StartGameWithoutHumans.new
 121     $paired = []
 122     $called = 0
 123     def @pairing.start_game(p1,p2)
 124       $called += 1
 125       $paired << [p1,p2]
 126     end
 127     @a = ShogiServer::BasicPlayer.new
 128     @a.name = "a"
 129     @a.win  = 1
 130     @a.loss = 2
 131     @a.rate = 0
 132     @b = ShogiServer::BasicPlayer.new
 133     @b.name = "b"
 134     @b.win  = 10
 135     @b.loss = 20
 136     @b.rate = 1500
 137     @c = ShogiServer::BasicPlayer.new
 138     @c.name = "c"
 139     @c.win  = 100
 140     @c.loss = 200
 141     @c.rate = 1000
 142     @d = ShogiServer::BasicPlayer.new
 143     @d.name = "d"
 144     @d.win  = 1000
 145     @d.loss = 2000
 146     @d.rate = 2000
 147     @e = ShogiServer::BasicPlayer.new
 148     @e.name = "e"
 149     @e.win  = 3000
 150     @e.loss = 3000
 151     @e.rate = 3000
 152     @f = ShogiServer::BasicPlayer.new
 153     @f.name = "f"
 154     @f.win  = 4000
 155     @f.loss = 4000
 156     @f.rate = 4000
 157     @g = ShogiServer::BasicPlayer.new
 158     @g.name = "g"
 159     @g.win  = 5000
 160     @g.loss = 5000
 161     @g.rate = 5000
 162     @h = ShogiServer::BasicPlayer.new
 163     @h.name = "h"
 164     @h.win  = 6000
 165     @h.loss = 6000
 166     @h.rate = 6000
 167   end
 168
 169   def test_match_one_player
 170     players = [@a]
 171     @pairing.match(players)
 172     assert_equal(0, $called)
 173   end
 174
 175   def test_match_one_player_human
 176     @a.name += "_human"
 177     players = [@a]
 178     @pairing.match(players)
 179     assert_equal(0, $called)
 180   end
 181
 182   def test_match_two_players
 183     players = [@a,@b]
 184     @pairing.match(players)
 185     assert_equal(1, $called)
 186   end
 187
 188   def test_match_two_players_humans
 189     @a.name += "_human"
 190     @b.name += "_human"
 191     players = [@a,@b]
 192     @pairing.match(players)
 193     assert_equal(1, $called)
 194   end
 195
 196   def test_match_zero_player
 197     players = []
 198     @pairing.match(players)
 199     assert_equal(0, $called)
 200   end
 201
 202   def test_match_three_players
 203     players = [@a,@b,@c]
 204     @pairing.match(players)
 205     assert_equal(1, $called)
 206   end
 207
 208   def test_match_three_players_a_human
 209     @a.name += "_human"
 210     players = [@a,@b,@c]
 211     @pairing.match(players)
 212     assert_equal(1, $called)
 213     assert_equal(1, players.size)
 214     assert_equal(@c, players[0])
 215   end
 216
 217   def test_match_three_players_b_human
 218     @b.name += "_human"
 219     players = [@a,@b,@c]
 220     @pairing.match(players)
 221     assert_equal(1, $called)
 222     assert_equal(1, players.size)
 223     assert_equal(@c, players[0])
 224   end
 225
 226   def test_match_three_players_c_human
 227     @c.name += "_human"
 228     players = [@a,@b,@c]
 229     @pairing.match(players)
 230     assert_equal(1, $called)
 231     assert_equal(1, players.size)
 232     assert_equal(@c, players[0])
 233   end
 234
 235   def test_match_three_players_ab_human
 236     @a.name += "_human"
 237     @b.name += "_human"
 238     players = [@a,@b,@c]
 239     @pairing.match(players)
 240     assert_equal(1, $called)
 241     assert_equal(1, players.size)
 242     assert_equal(@b, players[0])
 243   end
 244
 245   def test_match_three_players_bc_human
 246     @b.name += "_human"
 247     @c.name += "_human"
 248     players = [@a,@b,@c]
 249     @pairing.match(players)
 250     assert_equal(1, $called)
 251     assert_equal(1, players.size)
 252     assert_equal(@c, players[0])
 253   end
 254
 255   def test_match_four_players
 256     players = [@a,@b,@c,@d]
 257     @pairing.match(players)
 258     assert_equal(2, $called)
 259   end
 260
 261   def test_match_four_players_ab_human
 262     @a.name += "_human"
 263     @b.name += "_human"
 264     players = [@a,@b,@c,@d]
 265     @pairing.match(players)
 266     assert_equal(2, $paired.size)
 267     assert(same_pair?([@a,@c], $paired[0]))
 268     assert(same_pair?([@b,@d], $paired[1]))
 269   end
 270
 271   def test_match_four_players_bc_human
 272     @b.name += "_human"
 273     @c.name += "_human"
 274     players = [@a,@b,@c,@d]
 275     @pairing.match(players)
 276     assert_equal(2, $paired.size)
 277     assert(same_pair?([@a,@b], $paired[0]))
 278     assert(same_pair?([@c,@d], $paired[1]))
 279   end
 280
 281   def test_match_four_players_abc_human
 282     @a.name += "_human"
 283     @b.name += "_human"
 284     @c.name += "_human"
 285     players = [@a,@b,@c,@d]
 286     @pairing.match(players)
 287     assert_equal(2, $paired.size)
 288     assert(same_pair?([@a,@d], $paired[0]))
 289     assert(same_pair?([@b,@c], $paired[1]))
 290   end
 291
 292   def test_match_four_players_bcd_human
 293     @b.name += "_human"
 294     @c.name += "_human"
 295     @d.name += "_human"
 296     players = [@a,@b,@c,@d]
 297     @pairing.match(players)
 298     assert_equal(2, $paired.size)
 299     assert(same_pair?([@a,@c], $paired[0]))
 300     assert(same_pair?([@b,@d], $paired[1]))
 301   end
 302
 303   def test_match_four_players_abcd_human
 304     @a.name += "_human"
 305     @b.name += "_human"
 306     @c.name += "_human"
 307     @d.name += "_human"
 308     players = [@a,@b,@c,@d]
 309     @pairing.match(players)
 310     assert_equal(2, $paired.size)
 311     assert(same_pair?([@a,@b], $paired[0]))
 312     assert(same_pair?([@c,@d], $paired[1]))
 313   end
 314
 315   def test_match_eight_players_efgh_human
 316     @e.name += "_human"
 317     @f.name += "_human"
 318     @g.name += "_human"
 319     @h.name += "_human"
 320     players = [@a,@b,@c,@d,@e,@f,@g,@h]
 321     @pairing.match(players)
 322     assert_equal(4, $paired.size)
 323     assert(same_pair?([@e,@c], $paired[0]))
 324     assert(same_pair?([@d,@g], $paired[1]))
 325     assert(same_pair?([@a,@f], $paired[2]))
 326     assert(same_pair?([@b,@h], $paired[3]))
 327   end
 328 end
 329
 330 class TestLeastDiff < Test::Unit::TestCase
 331
 332   class MockLeague
 333     def initialize
 334       @players = []
 335     end
 336
 337     def add(player)
 338       @players << player
 339     end
 340
 341     def find(name)
 342       @players.find do |p|
 343         p.name == name
 344       end
 345     end
 346   end
 347
 348   def setup
 349     $league = MockLeague.new
 350
 351     @pairing= ShogiServer::LeastDiff.new
 352     $paired = []
 353     $called = 0
 354     def @pairing.start_game(p1,p2)
 355       $called += 1
 356       $paired << [p1,p2]
 357     end
 358
 359     @file = Pathname.new(File.join(File.dirname(__FILE__), "floodgate_history.yaml"))
 360     @history = ShogiServer::League::Floodgate::History.new @file
 361
 362     @a = ShogiServer::BasicPlayer.new
 363     @a.player_id = "a"
 364     @a.name = "a"
 365     @a.win  = 1
 366     @a.loss = 2
 367     @a.rate = 500
 368     @b = ShogiServer::BasicPlayer.new
 369     @b.player_id = "b"
 370     @b.name = "b"
 371     @b.win  = 10
 372     @b.loss = 20
 373     @b.rate = 800
 374     @c = ShogiServer::BasicPlayer.new
 375     @c.player_id = "c"
 376     @c.name = "c"
 377     @c.win  = 100
 378     @c.loss = 200
 379     @c.rate = 1000
 380     @d = ShogiServer::BasicPlayer.new
 381     @d.player_id = "d"
 382     @d.name = "d"
 383     @d.win  = 1000
 384     @d.loss = 2000
 385     @d.rate = 1500
 386     @e = ShogiServer::BasicPlayer.new
 387     @e.player_id = "e"
 388     @e.name = "e"
 389     @e.win  = 3000
 390     @e.loss = 3000
 391     @e.rate = 2000
 392     @f = ShogiServer::BasicPlayer.new
 393     @f.player_id = "f"
 394     @f.name = "f"
 395     @f.win  = 4000
 396     @f.loss = 4000
 397     @f.rate = 2150
 398     @g = ShogiServer::BasicPlayer.new
 399     @g.player_id = "g"
 400     @g.name = "g"
 401     @g.win  = 5000
 402     @g.loss = 5000
 403     @g.rate = 2500
 404     @h = ShogiServer::BasicPlayer.new
 405     @h.player_id = "h"
 406     @h.name = "h"
 407     @h.win  = 6000
 408     @h.loss = 6000
 409     @h.rate = 3000
 410     @x = ShogiServer::BasicPlayer.new
 411     @x.player_id = "x"
 412     @x.name = "x"
 413
 414     $league.add(@a)
 415     $league.add(@b)
 416     $league.add(@c)
 417     $league.add(@d)
 418     $league.add(@e)
 419     $league.add(@f)
 420     $league.add(@g)
 421     $league.add(@h)
 422     $league.add(@x)
 423   end
 424
 425   def teardown
 426     @file.delete if @file.exist?
 427   end
 428
 429   def assert_pairs(x_array, y_array)
 430     if (x_array.size != y_array.size)
 431       assert_equal(x_array.size, y_array.size)
 432       return
 433     end
 434     i = 0
 435
 436     if (x_array.size == 1)
 437       assert_equal(x_array[0].name, y_array[0].name)
 438       return
 439     end
 440
 441     ret = true
 442     while i < x_array.size
 443       if i == x_array.size-1
 444         assert_equal(x_array[i].name, y_array[i].name)
 445         break
 446       end
 447       px1 = x_array[i]
 448       px2 = x_array[i+1]
 449       py1 = y_array[i]
 450       py2 = y_array[i+1]
 451
 452       if ! ((px1.name == py1.name && px2.name == py2.name) ||
 453             (px1.name == py2.name && px2.name == py1.name))
 454         ret = false
 455       end
 456       i += 2
 457     end
 458
 459     assert(ret)
 460   end
 461
 462   def test_match_one_player
 463     players = [@a]
 464     assert_equal(0, @pairing.calculate_diff_with_penalty(players,nil))
 465     r = @pairing.match(players)
 466     assert_pairs([@a], r)
 467   end
 468
 469   def test_match_two_players
 470     players = [@a,@b]
 471     assert_equal(@b.rate-@a.rate, @pairing.calculate_diff_with_penalty([@a,@b],nil))
 472     assert_equal(@b.rate-@a.rate, @pairing.calculate_diff_with_penalty([@b,@a],nil))
 473     r = @pairing.match(players)
 474     assert_pairs([@a,@b], r)
 475   end
 476
 477   def test_match_three_players
 478     players = [@h,@a,@b]
 479     assert_equal(300,  @pairing.calculate_diff_with_penalty([@a,@b,@h],nil))
 480     assert_equal(2200, @pairing.calculate_diff_with_penalty([@b,@h,@a],nil))
 481     r = @pairing.match(players)
 482     assert_pairs([@a,@b,@h], r)
 483     assert_pairs([@a,@b,@h], players)
 484   end
 485
 486   def test_calculate_diff_with_penalty
 487     players = [@a,@b]
 488     assert_equal(@b.rate-@a.rate, @pairing.calculate_diff_with_penalty(players,nil))
 489
 490     dummy = nil
 491     def @history.make_record(game_result)
 492       {:game_id => "wdoor+floodgate-900-0-a-b-1",
 493        :black => "b",  :white => "a",
 494        :winner => "a", :loser => "b"}
 495     end
 496     @history.update(dummy)
 497     assert_equal(@b.rate-@a.rate+400, @pairing.calculate_diff_with_penalty(players, @history))
 498   end
 499
 500   def test_calculate_diff_with_penalty2
 501     players = [@a,@b,@g,@h]
 502     assert_equal(@b.rate-@a.rate+@h.rate-@g.rate, @pairing.calculate_diff_with_penalty(players,nil))
 503   end
 504
 505   def test_calculate_diff_with_penalty2_1
 506     players = [@a,@b,@g,@h]
 507     assert_equal(@b.rate-@a.rate+@h.rate-@g.rate, @pairing.calculate_diff_with_penalty(players,nil))
 508     dummy = nil
 509     def @history.make_record(game_result)
 510       {:game_id => "wdoor+floodgate-900-0-a-b-1",
 511        :black => "b",  :white => "a",
 512        :winner => "a", :loser => "b"}
 513     end
 514     @history.update(dummy)
 515     assert_equal(@b.rate-@a.rate+400+@h.rate-@g.rate, @pairing.calculate_diff_with_penalty(players, @history))
 516   end
 517
 518   def test_calculate_diff_with_penalty2_2
 519     players = [@a,@b,@g,@h]
 520     assert_equal(@b.rate-@a.rate+@h.rate-@g.rate, @pairing.calculate_diff_with_penalty(players,nil))
 521     dummy = nil
 522     def @history.make_record(game_result)
 523       {:game_id => "wdoor+floodgate-900-0-a-b-1",
 524        :black => "g",  :white => "h",
 525        :winner => "h", :loser => "g"}
 526     end
 527     @history.update(dummy)
 528     assert_equal(@b.rate-@a.rate+400+@h.rate-@g.rate, @pairing.calculate_diff_with_penalty(players, @history))
 529     #assert_equal(@b.rate-@a.rate+400+@h.rate-@g.rate+400, @pairing.calculate_diff_with_penalty(players, [@b,@a,@h,@g]))
 530   end
 531
 532   def test_calculate_diff_with_penalty2_3
 533     players = [@a,@b,@g,@h]
 534     assert_equal(@b.rate-@a.rate+@h.rate-@g.rate, @pairing.calculate_diff_with_penalty(players,nil))
 535     dummy = nil
 536     def @history.make_record(game_result)
 537       {:game_id => "wdoor+floodgate-900-0-a-b-1",
 538        :black => "g",  :white => "h",
 539        :winner => "h", :loser => "g"}
 540     end
 541     @history.update(dummy)
 542     def @history.make_record(game_result)
 543       {:game_id => "wdoor+floodgate-900-0-a-b-1",
 544        :black => "b",  :white => "a",
 545        :winner => "a", :loser => "b"}
 546     end
 547     @history.update(dummy)
 548     assert_equal(@b.rate-@a.rate+400+@h.rate-@g.rate+400, @pairing.calculate_diff_with_penalty(players, @history))
 549   end
 550
 551   def test_get_player_rate_0
 552     assert_equal(2150, @pairing.get_player_rate(@x, @history))
 553
 554     dummy = nil
 555     def @history.make_record(game_result)
 556       {:game_id => "wdoor+floodgate-900-0-x-a-1",
 557        :black => "x",  :white => "a",
 558        :winner => "x", :loser => "a"}
 559     end
 560     @history.update(dummy)
 561     assert_equal(@a.rate+100, @pairing.get_player_rate(@x, @history))
 562
 563     def @history.make_record(game_result)
 564       {:game_id => "wdoor+floodgate-900-0-x-b-1",
 565        :black => "x",  :white => "b",
 566        :winner => "b", :loser => "x"}
 567     end
 568     @history.update(dummy)
 569
 570     assert_equal((@a.rate+100+@b.rate-100)/2, @pairing.get_player_rate(@x, @history))
 571   end
 572 end
 573