def testSlidePairFatty(self):
b = Board(4,3)
#add a fatty to column 1 and another to column 0
fat1 = DummyPiece(2,2)
fat1.name = "fat right"
fat0 = DummyPiece(2,2)
fat0.name = "fat left"
fat0.slidePriority = 0
fat1.slidePriority = 0
b.addPiece(fat1, 1) #fat right is added first
b.addPiece(fat0, 0)
self.assertTrue(b.selfConsistent())
b.normalize()
#fat right (fat1) should be below fat left (fat0)
self.assertEqual(fat0.position, [2,0])
self.assertEqual(fat1.position, [0,1])
self.assertTrue(b.selfConsistent)
##make fat0 have priority
logging.debug("priority is about to be changed")
fat0.slidePriority = 5
b.normalize()
##fat0 should now be below fat1
self.assertEqual(fat0.position, [0,0])
self.assertEqual(fat1.position, [2,1])
self.assertTrue(b.selfConsistent)
def testSlideFatty(self):
b = Board(4, 4)
#add two small pieces in front of fatty
piece1 = DummyPiece(1, 1, chargeable = False)
piece2 = DummyPiece(1, 1, chargeable = False)
fatpiece = DummyPiece(2, 2, chargeable = False)
fatpiece.name = "fatty"
b.addPiece(piece1, 0)
b.addPiece(piece2, 0)
b.addPiece(fatpiece, 0)
b.normalize()
#the two squares (0,1) and (1,1) in front of fatty should be empty
self.assertEqual(b[0,1], None)
self.assertEqual(b[1,1], None)
#fatty should be at position (2, 0)
self.assertEqual(fatpiece.position, [2,0])
self.assertTrue(b.selfConsistent)
#now make fatty slide in front
fatpiece.slidePriority = 5
b.normalize()
#fatty should occupy the (0,0) position
self.assertEqual(fatpiece.position,[0,0])
self.assertTrue(b.selfConsistent)
def testTransform(self):
b = Board(6, 8)
pieces = [DummyPiece(1, 1, transformable = True) for c in range(4)]
for c in range(4):
b.addPiece(pieces[c], c)
transformedPieces = []
def transformHandler(p): transformedPieces.append(p)
b.wallMade.addHandler(transformHandler)
b.normalize()
self.assertEqual(len(transformedPieces[0][0]), 4)
def testPieceUpdate(self):
b = Board(3, 3)
piece1 = DummyPiece(1, 1)
piece2 = DummyPiece(1, 1)
piece3 = DummyPiece(1, 1)
piece3.slidePriority = 5
updates = [None]
def updateHandler(p): updates[0] = p
b.pieceUpdated.addHandler(updateHandler)
b.addPiece(piece1, 0)
b.normalize()
self.assertEqual(updates[0], set([piece1]))
b.addPiece(piece2, 1)
b.normalize()
self.assertEqual(updates[0], set([piece2]))
b.addPiece(piece3, 0)
b.normalize()
self.assertEqual(updates[0], set([piece1, piece3]))
b.deletePiece(piece1)
self.assertEqual(updates[0], set([piece1]))