import unittest from scripts.balance_lab import ( aggregate_class_abilities, aggregate_class_specs, automatic_parties, coverage_parties, random_party, search_bosses, tuning_label, ) class BalanceLabControllerTests(unittest.TestCase): def test_random_party_is_repeatable_and_complete(self) -> None: builds = [ {"id": "tank-a", "role": "tank"}, {"id": "tank-b", "role": "tank"}, {"id": "healer-a", "role": "healer"}, {"id": "healer-b", "role": "healer"}, *({"id": f"damage-{index}", "role": "damage"} for index in range(6)), ] first = random_party(builds, 17) second = random_party(builds, 17) self.assertEqual(first, second) self.assertEqual([member["role"] for member in first], [ "tank", "healer", "damage", "damage", "damage", ]) self.assertEqual(len({member["id"] for member in first[2:]}), 3) def test_tuning_flags_keep_both_sides_of_the_median(self) -> None: self.assertEqual(tuning_label(1.51), "very high") self.assertEqual(tuning_label(1.25), "high") self.assertEqual(tuning_label(1.0), "baseline") self.assertEqual(tuning_label(0.75), "low") self.assertEqual(tuning_label(0.49), "very low") self.assertEqual(tuning_label(None), "no baseline") def test_automatic_parties_are_unique_repeatable_and_source_isolated(self) -> None: builds = [] for source in ("wow", "coa"): builds.extend([ {"id": f"{source}-tank-a", "source": source, "role": "tank"}, {"id": f"{source}-tank-b", "source": source, "role": "tank"}, {"id": f"{source}-healer-a", "source": source, "role": "healer"}, {"id": f"{source}-healer-b", "source": source, "role": "healer"}, *({"id": f"{source}-damage-{index}", "source": source, "role": "damage"} for index in range(6)), ]) first = automatic_parties(builds, ("wow", "coa"), 4, 29) second = automatic_parties(builds, ("wow", "coa"), 4, 29) self.assertEqual(first, second) self.assertEqual(len(first), 8) self.assertEqual(len({configuration["id"] for configuration in first}), 8) for configuration in first: self.assertTrue(all(member["source"] == configuration["source"] for member in configuration["party"])) self.assertEqual([member["role"] for member in configuration["party"]], [ "tank", "healer", "damage", "damage", "damage", ]) def test_coverage_parties_include_every_build_with_minimum_unique_parties(self) -> None: builds = [ {"id": "tank-a", "source": "wow", "role": "tank"}, {"id": "tank-b", "source": "wow", "role": "tank"}, {"id": "healer-a", "source": "wow", "role": "healer"}, {"id": "healer-b", "source": "wow", "role": "healer"}, *({"id": f"damage-{index}", "source": "wow", "role": "damage"} for index in range(7)), ] first = coverage_parties(builds, ("wow",), None, 19) second = coverage_parties(builds, ("wow",), None, 19) self.assertEqual(first, second) self.assertEqual(len(first), 3) covered = {member["id"] for configuration in first for member in configuration["party"]} self.assertEqual(covered, {build["id"] for build in builds}) signatures = {tuple(member["id"] for member in configuration["party"]) for configuration in first} self.assertEqual(len(signatures), 3) for configuration in first: self.assertEqual([member["role"] for member in configuration["party"]], [ "tank", "healer", "damage", "damage", "damage", ]) self.assertEqual(len({member["id"] for member in configuration["party"]}), 5) def test_specific_coverage_uses_fillers_but_guarantees_selected_builds(self) -> None: builds = [ {"id": "tank-a", "source": "wow", "role": "tank"}, {"id": "tank-b", "source": "wow", "role": "tank"}, {"id": "healer-a", "source": "wow", "role": "healer"}, {"id": "healer-b", "source": "wow", "role": "healer"}, *({"id": f"damage-{index}", "source": "wow", "role": "damage"} for index in range(5)), ] selected = ["tank-a", "tank-b", "damage-4"] configurations = coverage_parties(builds, ("wow",), selected, 7) self.assertEqual(len(configurations), 2) observed = [member["id"] for configuration in configurations for member in configuration["party"]] covered = set(observed) self.assertTrue(set(selected).issubset(covered)) self.assertTrue(all(observed.count(build_id) == 1 for build_id in selected)) def test_boss_search_matches_name_dungeon_key_and_source(self) -> None: bosses = [ {"key": "gundrak/eck", "name": "Eck the Ferocious", "dungeonTitle": "Gundrak", "category": "wow"}, {"key": "abbey/duke", "name": "Duke", "dungeonTitle": "Forsaken Abbey", "category": "rom"}, ] self.assertEqual([boss["key"] for boss in search_bosses("ferocious", bosses)], ["gundrak/eck"]) self.assertEqual([boss["key"] for boss in search_bosses("forsaken", bosses)], ["abbey/duke"]) self.assertEqual([boss["key"] for boss in search_bosses("rom", bosses)], ["abbey/duke"]) def test_class_spec_aggregate_keeps_damage_and_healing_received_metrics(self) -> None: key = ("wow", "priest-holy", "priest", "Priest", "Holy", "healer") rows = aggregate_class_specs({key: [ { "partyConfigurationId": "wow-party-001", "elapsedSeconds": 10, "damage": 200, "bossDamage": 180, "healing": 500, "healingReceived": 300, "damageTaken": 120, "absorbs": 25, "deathEvents": 0, }, { "partyConfigurationId": "wow-party-002", "elapsedSeconds": 20, "damage": 400, "bossDamage": 360, "healing": 900, "healingReceived": 600, "damageTaken": 330, "absorbs": 50, "deathEvents": 1, }, ]}) self.assertEqual(len(rows), 1) self.assertEqual(rows[0]["dtps"], 15) self.assertEqual(rows[0]["totalDamageTaken"], 450) self.assertEqual(rows[0]["hpsReceived"], 30) self.assertEqual(rows[0]["totalHealingReceived"], 900) def test_ability_aggregate_keeps_used_and_available_but_unused_rows(self) -> None: class_key = ("wow", "mage-fire", "mage", "Mage", "Fire", "damage") class_runs = [ {"partyConfigurationId": "party-1", "bossKey": "boss-1", "elapsedSeconds": 10, "damage": 100}, {"partyConfigurationId": "party-2", "bossKey": "boss-2", "elapsedSeconds": 10, "damage": 100}, ] base = { "abilityName": "Fireball", "icon": "/fireball.png", "categories": ["damage"], "cooldownSeconds": 0, "available": True, "casts": 0, "hits": 0, "rawDamage": 0, "effectiveDamage": 0, "rawHealing": 0, "effectiveHealing": 0, "overheal": 0, "absorbs": 0, } groups = { (*class_key, "mage-fireball"): [ {**base, "partyConfigurationId": "party-1", "bossKey": "boss-1", "elapsedSeconds": 10, "used": True, "casts": 1, "hits": 1, "rawDamage": 120, "effectiveDamage": 100}, {**base, "partyConfigurationId": "party-2", "bossKey": "boss-2", "elapsedSeconds": 10, "used": False}, ], (*class_key, "mage-unused-blast"): [ {**base, "abilityName": "Unused Blast", "partyConfigurationId": "party-1", "bossKey": "boss-1", "elapsedSeconds": 10, "used": False}, {**base, "abilityName": "Unused Blast", "partyConfigurationId": "party-2", "bossKey": "boss-2", "elapsedSeconds": 10, "used": False}, ], } rows = aggregate_class_abilities(groups, {class_key: class_runs}) by_id = {row["abilityId"]: row for row in rows} self.assertEqual(by_id["mage-fireball"]["activeRuns"], 1) self.assertEqual(by_id["mage-fireball"]["usageRate"], 0.5) self.assertEqual(by_id["mage-fireball"]["damageShare"], 0.5) self.assertTrue(by_id["mage-unused-blast"]["unusedAvailable"]) if __name__ == "__main__": unittest.main()