added unittest for seeding gamestate generation

fixed bug in gamestate.get_logistic_plan
This commit is contained in:
DuSack1220 2022-10-01 16:50:49 +02:00
parent f03b9d51b5
commit 6946ce9957
2 changed files with 19 additions and 0 deletions

View file

@ -77,6 +77,8 @@ Logistic Centers: {self.logistic_center}"""
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_plan = [x + 10 for x in self.logistic_plan]
def get_heuballen(self) -> int:
return self.heu_color

17
client_s2/test.py Normal file
View 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()