def testGetPayoffMatrixFromFile(self):
games = {
'bosx.nfg' :
[[[4, 2], [0, 0], [0, 1]],
[[0, 0], [2, 4], [1, 3]]],
'sspes.nfg' :
[[[0, 0], [-1, 1], [1, -1], [1, -1], [-1, 1]],
[[1, -1], [0, 0], [-1, 1], [1, -1], [-1, 1]],
[[-1, 1], [1, -1], [0, 0], [-1, 1], [1, -1]],
[[-1, 1], [-1, 1], [1, -1], [0, 0], [1, -1]],
[[1, -1], [1, -1], [-1, 1], [-1, 1], [0, 0]]],
'airstrike4.nfg' :
[[[0, 0], [2, -2], [2, -2], [2, -2]],
[[3, -3], [0, 0], [3, -3], [3, -3]],
[[4, -4], [4, -4], [0, 0], [4, -4]],
[[5, -5], [5, -5], [5, -5], [0, 0]]],
'bos.nfg' :
[[[2, 1], [0, 0]],
[[0, 0], [1, 2]]],
'sspb.nfg' :
[[[0, 0], [-1, 1], [1, -1], [-1, 1]],
[[1, -1], [0, 0], [-1, 1], [-1, 1]],
[[-1, 1], [1, -1], [0, 0], [1, -1]],
[[1, -1], [1, -1], [-1, 1], [0, 0]]],
'chicken.nfg' :
[[[3, 3], [2, 4]],
[[4, 2], [1, 1]]],
'coordination.nfg' :
[[[3, 3], [0, 0]],
[[0, 0], [1, 1]]],
'airstrike.nfg' :
[[[0, 0], [2, -2], [2, -2]],
[[3, -3], [0, 0], [3, -3]],
[[4, -4], [4, -4], [0, 0]]]
}
for (filename, payoffMatrix) in games.iteritems():
game = NfgReader().read(filename="../games/" + filename)
self.assertEquals(payoffMatrix, game.getPayoffMatrix())
