def test_generate_matrix_2(self):
r = StringIO.StringIO("1 1\n1 1 1")
matrix = pfd_generate_matrix(r)
self.assert_(matrix != False)
self.assert_(matrix == [[1, 0], [1, 1]])
def test_generate_matrix_3(self):
r = StringIO.StringIO("3 3\n2 1 1\n3 2 3 2\n 1 1 2")
matrix = pfd_generate_matrix(r)
self.assert_(matrix != False)
self.assert_(matrix == [[3, 0, 0, 0], [1, 0, 1, 0], [1, 1, 0, 0], [2, 0, 1, 1]])
def test_generate_matrix_1(self):
r = StringIO.StringIO("2 1\n2 1 1")
matrix = pfd_generate_matrix(r)
self.assert_(matrix != False)
self.assert_(matrix == [[2, 0, 0], [0, 0, 0], [1, 1, 0]])
