def test_clear_2 (self) :
r = StringIO.StringIO("2 1\n1 1 2")
a = [0, 0]
c = [0]
b = pfd_initialize(r, a, c)
matrix = getMatrix()
self.assert_(b == True)
self.assert_(a[0] == 2)
self.assert_(a[1] == 1)
self.assert_(matrix == [[0, 1],[0, 0]] )
pfd_clear(1)
matrix = getMatrix()
self.assert_(matrix == [[0, 0],[0, 0]] )
def test_clear_1 (self) :
r = StringIO.StringIO("5 4\n3 2 1 5\n2 2 5 3\n4 1 3\n5 1 1")
a = [0, 0]
c = [0]
b = pfd_initialize(r, a, c)
matrix = getMatrix()
self.assert_(b == True)
self.assert_(a[0] == 5)
self.assert_(a[1] == 4)
self.assert_(matrix == [[0, 0, 0, 0, 0],
[0, 0, 1, 0, 1],
[1, 0, 0, 0, 1],
[0, 0, 1, 0, 0],
[1, 0, 0, 0, 0]] )
pfd_clear(0)
matrix = getMatrix()
self.assert_(matrix == [[0, 0, 0, 0, 0],
[0, 0, 1, 0, 1],
[0, 0, 0, 0, 1],
[0, 0, 1, 0, 0],
[0, 0, 0, 0, 0]] )
