18 lines
440 B
Python
18 lines
440 B
Python
|
|
||
|
|
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),
|
||
|
|
(3, 3)):
|
||
|
|
boxes = generate_colour_boxes(*args)
|
||
|
|
print(boxes)
|
||
|
|
self.assertTrue(makes_all_wells_white(boxes))
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
unittest.main()
|