class TestGoldHeuristicValue(unittest.TestCase):
def setUp(self):
self.heuristic = EloGoldHeuristic()
def test_heuristic_value_yupgvpr5_2400(self):
"""Test the heuristic value on yupgvpr5 2400"""
# Build models
status_dict = get_status_samples_dict()["yupgvpr5"][2400]
map_obj = Map(status_dict["game"]["board"]["tiles"])
status = Status(status_dict["game"], map_obj)
heuristic_tuple = self.heuristic.heuristic(status)
expected_tuple = (8.0, -8.0, -8.0, 8.0)
self.assertEqual(heuristic_tuple, expected_tuple)
def test_heuristic_value_nztclzmi_0(self):
"""Test the heuristic value on nztclzmi 0"""
# Build models
status_dict = get_status_samples_dict()["nztclzmi"][0]
map_obj = Map(status_dict["game"]["board"]["tiles"])
status = Status(status_dict["game"], map_obj)
heuristic_tuple = self.heuristic.heuristic(status)
self.assertEqual(heuristic_tuple, (0, 0, 0, 0))
class TestEloGoldHeuristicOnLeafs(unittest.TestCase):
def setUp(self):
self.heuristic = EloGoldHeuristic()
def perform_test(self, status_dict):
"""Test that the heuristic on leafs is consistent with the utility"""
# Build models
map_obj = Map(status_dict["game"]["board"]["tiles"])
status = Status(status_dict["game"], map_obj)
# Simulate a final status
status.turn = status.max_turns
heuristic_tuple = self.heuristic.heuristic(status)
utility_tuple = utility(status)
# Check that heuristic and utility are consistent
for i in range(4):
for j in range(4):
if utility_tuple[i] > utility_tuple[j]:
self.assertGreater(heuristic_tuple[i], heuristic_tuple[j])
elif utility_tuple[i] == utility_tuple[j]:
self.assertAlmostEqual(
heuristic_tuple[i],
heuristic_tuple[j]
)
else:
self.assertLess(heuristic_tuple[i], heuristic_tuple[j])
tests = get_status_samples()
class TestHeroScoreCodomain(unittest.TestCase):
def setUp(self):
self.heuristic = EloGoldHeuristic()
def perform_test(self, status_dict):
"""Test that the values of gold_score are between 0 and 1"""
# Build models
map_obj = Map(status_dict["game"]["board"]["tiles"])
status = Status(status_dict["game"], map_obj)
for hero_id in range(4):
for opponent_id in range(4):
score = self.heuristic.hero_score(status, hero_id, opponent_id)
self.assertGreaterEqual(score, 0)
self.assertLessEqual(score, 1)
tests = get_status_samples()
class TestEloGoldHeuristicZeroSum(unittest.TestCase):
def setUp(self):
self.heuristic = EloGoldHeuristic()
def perform_test(self, status_dict):
"""Test that the heuristic is a zero sum tuple"""
# Build models
map_obj = Map(status_dict["game"]["board"]["tiles"])
status = Status(status_dict["game"], map_obj)
heuristic_tuple = self.heuristic.heuristic(status)
self.assertAlmostEqual(
sum(heuristic_tuple), 0,
msg="heuristic is {h}, but the sum is non-zero: {sum}".format(
h=heuristic_tuple,
sum=sum(heuristic_tuple)
)
)
tests = get_status_samples()
def setUp(self):
self.heuristic = EloGoldHeuristic()
