#format python
"""BadukSimulation by yong27
"""
import unittest

class GoMap:
    def __init__(self, line=19):
        self.indices = range(line)
        self.map = [[0]*line for i in self.indices]

    def position(self, (i,j)):
        assert i in self.indices and j in self.indices, "Index out of range"
        return self.map[i][j]

    def hand(self, color, (i,j)):
        assert not self.map[i][j], "Can't select handed position"
        self.map[i][j] = color
        self.checkStatus()

    def handBlack(self, (i,j)):
        self.hand(1, (i,j))
    def handWhite(self, (i,j)):
        self.hand(2, (i,j))

    def checkStatus(self):
        pass

    def getAround(self,(i,j)):
        return dict(U=(i-1,j), D=(i+1,j), L=(i,j-1), R=(i,j+1))

    def searchAround(self, (i,j)):
        result = dict()
        around = self.getAround((i,j))
        for mark in around:
            try:
                result[mark] = self.position(around[mark])
            except AssertionError:
                pass
        return result

    def getSwarm(self, (i,j)):
        assert self.position((i,j))
        sw = Swarm(self.position((i,j)), (i,j))
        inputs = [(i,j),]
        while inputs:
            i,j = inputs[0]
            search = self.searchAround((i,j))
            for mark in search:
                linkedPos = self.getAround((i,j))[mark]
                if (search[mark] and search[mark] == self.position((i,j)) 
                        and linkedPos not in sw.cells):
                    sw.addCell(linkedPos)
                    inputs.append(linkedPos)
            inputs.remove((i,j))
        return sw

    def __repr__(self):
        pix = lambda p: p==1 and 'X' or p==2 and 'O' or '+'
        return '\n'.join([' '.join(map(pix,row)) for row in self.map])


class Swarm:
    def __init__(self, color, (i,j)):
        self.color = color
        self.cells = set()
        self.addCell((i,j))

    def addCell(self, (i,j)):
        self.cells.add((i,j))


class TestGoMap(unittest.TestCase):
    def testRepr(self):
        gm = GoMap()
        gm.handBlack((3,4))
        gm.handWhite((6,7))
        expected="""\
+ + + + + + + + + + + + + + + + + + +
+ + + + + + + + + + + + + + + + + + +
+ + + + + + + + + + + + + + + + + + +
+ + + + X + + + + + + + + + + + + + +
+ + + + + + + + + + + + + + + + + + +
+ + + + + + + + + + + + + + + + + + +
+ + + + + + + O + + + + + + + + + + +
+ + + + + + + + + + + + + + + + + + +
+ + + + + + + + + + + + + + + + + + +
+ + + + + + + + + + + + + + + + + + +
+ + + + + + + + + + + + + + + + + + +
+ + + + + + + + + + + + + + + + + + +
+ + + + + + + + + + + + + + + + + + +
+ + + + + + + + + + + + + + + + + + +
+ + + + + + + + + + + + + + + + + + +
+ + + + + + + + + + + + + + + + + + +
+ + + + + + + + + + + + + + + + + + +
+ + + + + + + + + + + + + + + + + + +
+ + + + + + + + + + + + + + + + + + +"""
        self.assertEquals(expected, repr(gm))

    def testRaisedWhenHanded(self):
        gm = GoMap()
        gm.handBlack((3,4))
        self.assertRaises(AssertionError, gm.handWhite, (3,4))

    def testSearchAround(self):
        gm = GoMap()
        gm.handBlack((3,4))
        gm.handWhite((3,3))
        self.assertEquals(dict(U=0,D=0,L=0,R=1), gm.searchAround((3,3)))
        self.assertEquals(dict(U=0,D=0,L=2,R=0), gm.searchAround((3,4)))
        gm.handWhite((0,3))
        self.assertEquals(dict(D=0,L=0,R=2), gm.searchAround((0,2)))
        self.assertEquals(dict(D=0,R=0), gm.searchAround((0,0)))

    def testGetSwarm(self):
        gm = GoMap()
        gm.handBlack((3,4))
        self.assertEquals(set([(3,4)]), gm.getSwarm((3,4)).cells)
        gm.handBlack((3,3))
        self.assertEquals(set([(3,3),(3,4)]), gm.getSwarm((3,4)).cells)
        gm.handBlack((3,2))
        gm.handBlack((2,2))
        gm.handBlack((1,2))
        self.assertEquals(set([(3,3),(3,2),(2,2),(1,2),(3,4)]), gm.getSwarm((3,4)).cells)
        self.assertRaises(AssertionError, gm.getSwarm, (4,4))

    def _testCheckStatus(self):
        gm = GoMap()
        gm.handBlack((3,3))
        gm.handBlack((4,2))
        gm.handWhite((4,3))
        gm.handBlack((4,4))
        gm.handBlack((5,3))
        self.assertEquals(0, gm.position((4,3)))
        self.assertEquals(1, gm.whiteDead)

if __name__=='__main__':
    unittest.main(argv=('','-v'))