a-robots-conundrum/tests/colourboxes_tests.py

21 lines
521 B
Python
Raw Normal View History

2012-08-08 14:10:10 +02:00
from __future__ import print_function
import unittest
from robotgame.logic.colourboxes import *
class ColourboxesTest(unittest.TestCase):
def test_generation(self):
for args in ((1, 1),
(2, 2),
2012-08-08 14:12:38 +02:00
(3, 3),
(4, 6)):
2012-08-08 14:10:10 +02:00
boxes = generate_colour_boxes(*args)
2012-08-08 14:12:38 +02:00
print()
for box in boxes:
print(box)
2012-08-08 14:10:10 +02:00
self.assertTrue(makes_all_wells_white(boxes))
if __name__ == '__main__':
unittest.main()