/
gamebuildernp_test.py
85 lines (67 loc) · 3.81 KB
/
gamebuildernp_test.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
import unittest
import numpy as np
from gamebuildernp import PlayerSpecification, Strategy
from gamebuildernp import BayesianGame
class SampleGame(BayesianGame):
"""
This is the sample bayesian game contained in Chapter 7 of Essentials of Game Theory by Leyton-Brown.
"""
def __init__(self):
super(SampleGame, self).__init__(
game_name="Sample_Game",
player_specification=PlayerSpecification(player_types=[1, 2], player_actions=["U", "D"]),
opponent_specification=PlayerSpecification(player_types=[1, 2],
player_actions=["R",
"L"]))
self.type_probability = {(1, 1): .3,
(1, 2): .1,
(2, 1): .2,
(2, 2): .4}
self.utility_values = {(1, "U", 1, "L"): (2, 0),
(1, "U", 2, "L"): (2, 2),
(2, "U", 1, "L"): (2, 2),
(2, "U", 2, "L"): (2, 1),
(1, "U", 1, "R"): (0, 2),
(1, "U", 2, "R"): (0, 3),
(2, "U", 1, "R"): (0, 0),
(2, "U", 2, "R"): (0, 0),
(1, "D", 1, "L"): (0, 2),
(1, "D", 2, "L"): (3, 0),
(2, "D", 1, "L"): (0, 0),
(2, "D", 2, "L"): (0, 0),
(1, "D", 1, "R"): (2, 0),
(1, "D", 2, "R"): (1, 1),
(2, "D", 1, "R"): (1, 1),
(2, "D", 2, "R"): (1, 2)}
def get_types_probability(self, player_type, opponent_type):
return self.type_probability[(player_type, opponent_type)]
def get_utility(self, player_type, player_strategy, opponent_type, opponent_strategy):
player_action = player_strategy.get_action_by_type(player_type)
opponent_action = opponent_strategy.get_action_by_type(opponent_type)
result = self.utility_values[(player_type, player_action, opponent_type, opponent_action)]
return result
class BayesianGameTest(unittest.TestCase):
def __init__(self, *args, **kwargs):
super(BayesianGameTest, self).__init__(*args, **kwargs)
self.sample_game = SampleGame()
def test_pure_strategies(self):
expected_strategies = np.array([["U", "U"], ["U", "D"], ["D", "U"], ["D", "D"]])
actual_strategies = self.sample_game.player_specification.get_pure_strategies()
self.assertTrue(np.array_equal(actual_strategies, expected_strategies),
"expected: " + str(expected_strategies) +
" actual: " + str(actual_strategies))
expected_strategies = np.array([["R", "R"], ["R", "L"], ["L", "R"], ["L", "L"]])
actual_strategies = self.sample_game.opponent_specification.get_pure_strategies()
self.assertTrue(np.array_equal(actual_strategies, expected_strategies),
"expected: " + str(expected_strategies) +
" actual: " + str(actual_strategies))
def test_get_utility(self):
expected_opponent_utility = 1.
player_strategy = ("U", "U")
opponent_strategy = ("L", "L")
actual_player_utility, actual_opponent_utility = self.sample_game.get_expected_utilities(
(Strategy(player_strategy, self.sample_game.player_specification),
Strategy(opponent_strategy, self.sample_game.opponent_specification)))
self.assertAlmostEqual(actual_opponent_utility, expected_opponent_utility)
def test_get_game_file(self):
self.sample_game.get_strategic_game_format()