def testProblemGetDirichlet(self):
width = 3
height = 3
[nodes, boundary_nodes, tris] = generateRectangularMesh((width, height), (0,0), (1,1))
p = Problem(nodes, boundary_nodes, tris)
data = p.getDirichletData(lambda x,y: x*y)
self.assertEqual(data[0], 0)
self.assertEqual(4 in data, True)
self.assertEqual(2 in data, True)
