def test_seed_player_in_1v1(self):
new_player = generate_player_data("1v1", 2479)
teams = {"team 1": [generate_player_data("1v1", 2500)], "team 2": []}
algorithm = SimpleAverage(1)
added, updated_teams = algorithm.seed_player(new_player, teams)
self.assertTrue(added)
self.assertEqual(len(updated_teams["team 1"]), 1)
self.assertEqual(len(updated_teams["team 2"]), 1)
self.assertIn(new_player, updated_teams["team 2"])
def test_seed_player_in_1v1_doesnt_do_changes(self):
new_player = generate_player_data("1v1", 2500)
teams = {"team 1": [generate_player_data("1v1", 2900)], "team 2": []}
algorithm = SimpleAverage(1)
added, updated_teams = algorithm.seed_player(new_player, teams)
self.assertFalse(added)
self.assertEqual(len(updated_teams["team 1"]), 1)
self.assertEqual(len(updated_teams["team 2"]), 0)
self.assertNotIn(new_player, updated_teams["team 1"])
def test_game_mode_seeds_a_player_into_existing_teams(self):
game_mode = GameMode('duel', 2, 1, 'SimpleAverage')
new_player = generate_player_data("1v1", 2479)
teams = {"team 1": [generate_player_data("1v1", 2500)], "team 2": []}
added, is_filled, updated_teams = game_mode.seed_player(
new_player, teams)
self.assertTrue(added)
self.assertTrue(is_filled)
self.assertEqual(len(updated_teams["team 1"]), 1)
self.assertEqual(len(updated_teams["team 2"]), 1)
self.assertIn(new_player, updated_teams["team 2"])
def test_seed_player_in_team_deathmatch_doesnt_do_changes(self):
new_player = generate_player_data("team-deathmatch", 2000)
teams = {
"team 1": [
generate_player_data("team-deathmatch", 2500),
],
"team 2": [
generate_player_data("team-deathmatch", 2512),
]
}
algorithm = SimpleAverage(2)
added, updated_teams = algorithm.seed_player(new_player, teams)
self.assertFalse(added)
self.assertEqual(len(updated_teams["team 1"]), 1)
self.assertEqual(len(updated_teams["team 2"]), 1)
self.assertNotIn(new_player, updated_teams["team 1"])
self.assertNotIn(new_player, updated_teams["team 2"])
def test_seed_player_in_team_deathmatch(self):
new_player = generate_player_data("team-deathmatch", 2491)
teams = {
"team 1": [
generate_player_data("team-deathmatch", 2500),
generate_player_data("team-deathmatch", 2498),
],
"team 2": [
generate_player_data("team-deathmatch", 2512),
]
}
algorithm = SimpleAverage(2)
added, updated_teams = algorithm.seed_player(new_player, teams)
self.assertTrue(added)
self.assertEqual(len(updated_teams["team 1"]), 2)
self.assertEqual(len(updated_teams["team 2"]), 2)
self.assertIn(new_player, updated_teams["team 2"])
def test_seed_player_in_team_deathmatch_with_empty_groups(self):
new_player = generate_player_data("team-deathmatch", 2500)
rating_groups = {"team 1": [], "team 2": []}
algorithm = SimpleAverage(2)
added, updated_teams = algorithm.seed_player(new_player, rating_groups)
self.assertTrue(added)
self.assertEqual(len(updated_teams["team 1"]), 1)
self.assertEqual(len(updated_teams["team 2"]), 0)
self.assertIn(new_player, updated_teams["team 1"])
def test_game_mode_seeds_a_player_and_created_new_teams(self):
game_mode = GameMode('duel', 2, 1, 'SimpleAverage')
new_player = generate_player_data('find-opponent', 2500)
added, is_filled, updated_teams = game_mode.seed_player(new_player, {})
self.assertTrue(added)
self.assertFalse(is_filled)
self.assertEqual({"team 1", "team 2"}, set(updated_teams.keys()))
self.assertEqual(len(updated_teams["team 1"]), 1)
self.assertEqual(len(updated_teams["team 2"]), 0)
self.assertIn(new_player, updated_teams["team 1"])