Added colour box generation.

tests/colourboxes_tests.py

@@ -0,0 +1,17 @@
+
+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()