added unittest for seeding gamestate generation
fixed bug in gamestate.get_logistic_plan
This commit is contained in:
parent
f03b9d51b5
commit
6946ce9957
2 changed files with 19 additions and 0 deletions
|
@ -77,6 +77,8 @@ Logistic Centers: {self.logistic_center}"""
|
||||||
for i in range(0, len(self.logistic_plan) - 1):
|
for i in range(0, len(self.logistic_plan) - 1):
|
||||||
self.logistic_center[self.logistic_plan[i]][self.logistic_plan[i + 1]] += 1
|
self.logistic_center[self.logistic_plan[i]][self.logistic_plan[i + 1]] += 1
|
||||||
|
|
||||||
|
self.logistic_plan = [x + 10 for x in self.logistic_plan]
|
||||||
|
|
||||||
def get_heuballen(self) -> int:
|
def get_heuballen(self) -> int:
|
||||||
return self.heu_color
|
return self.heu_color
|
||||||
|
|
||||||
|
|
17
client_s2/test.py
Normal file
17
client_s2/test.py
Normal file
|
@ -0,0 +1,17 @@
|
||||||
|
import unittest
|
||||||
|
import compLib.Seeding as Seeding
|
||||||
|
|
||||||
|
|
||||||
|
class SeedingTest(unittest.TestCase):
|
||||||
|
def test_basic_seed(self):
|
||||||
|
gamestate = Seeding.Gamestate(0)
|
||||||
|
self.assertEqual(gamestate.seed, 0)
|
||||||
|
self.assertEqual(gamestate.heu_color, 1)
|
||||||
|
self.assertEqual(gamestate.get_heuballen(), 1)
|
||||||
|
self.assertEqual(gamestate.get_logistic_plan(),
|
||||||
|
[12, 13, 10, 13, 12, 10, 11, 10, 12, 11, 12, 13, 10, 12, 10, 11, 13, 11, 13, 11, 12])
|
||||||
|
self.assertEqual(gamestate.get_material_deliveries(), [[3, 1], [0, 3], [3, 1], [3, 1]])
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == '__main__':
|
||||||
|
unittest.main()
|
Reference in a new issue