mazeTests.py

import unittest
from PIL import Image
import ProbablyPython

class mazeUnitTests(unittest.TestCase):

    def test_pointException(self):
        with self.assertRaises(Exception) as context:
            ProbablyPython.Point("0", "1")
        self.assertTrue("Can not pass type string to class Point", str(context.exception))

    def test_setNextColor(self):
        inputs = {"sizeX": 51, "sizeY": 51, "shouldSolve": None, "solveColor": 1, "isGif": None}
        foo = [0,0,0,255]
        color = ProbablyPython.setNextColor(1, [255,0,0,255], [0,0,255,255], foo, foo, inputs)
        color = ProbablyPython.setNextColor(1, color, [0,0,255,255], foo, foo, inputs)
        self.assertEqual([0,0,255,255], color)

    def test_setEntranceExit(self):
        point = ProbablyPython.Point
        inputs = {"sizeX": 51, "sizeY": 51, "shouldSolve": None, "solveColor": 1, "isGif": None}
        myimage = Image.new('RGBA', (inputs['sizeX'],inputs['sizeY']), color=(0,0,0,255))
        img = myimage.load()
        coordinates = {'pos': point(1, 1), 'entrancePos': point(1, 1),
                       'exitPos': point(inputs['sizeX'] - 2, inputs['sizeY'] - 2)}
        ProbablyPython.setEntranceExit(img, coordinates, [0,0,255,255], [0,0,0,255], inputs)
        self.assertFalse(coordinates['entrancePos'].x > 1 and coordinates['entrancePos'].x < inputs['sizeX'] - 2
                         and coordinates['entrancePos'].y > 1 and coordinates['entrancePos'].y < inputs['sizeY'] - 2)
        self.assertFalse(coordinates['exitPos'].x > 1 and coordinates['exitPos'].x < inputs['sizeX'] - 2
                         and coordinates['exitPos'].y > 1 and coordinates['exitPos'].y < inputs['sizeY'] - 2)


class mazeIntegrationTests(unittest.TestCase):
    def test_mazeGen(self):
        data = {"sizeX": 51, "sizeY": 51, "shouldSolve": None, "solveColor": None, "isGif": None}
        image = ProbablyPython.generateMaze(data)
        img = image.load()
        testArray = []
        for x in range(image.size[0] - 1):
            for y in range(image.size[1] - 1):
                if x % 2 != 0 and y % 2 != 0:
                    testArray.append(img[x, y])
                    self.assertEqual(img[x, y], (255, 255, 255, 255))

    def test_mazeGenWithSolution(self):
        inputsRecieved = {"sizeX": 51, "sizeY": 51, "shouldSolve": True, "solveColor": 1, "isGif": None}
        image = ProbablyPython.generateMaze(inputsRecieved)
        img = image.load()
        testArray = []
        for x in range(image.size[0]-1):
            for y in range(image.size[1]-1):
                if x % 2 != 0 and y % 2 != 0:
                    testArray.append(img[x,y])
                    self.assertNotEqual(img[x,y], (0,0,0,255))


if __name__ == '__main__':
    unittest.main()