def testProblemStiffness2(self):
width = 4
height = 4
[nodes, boundary_nodes, tris] = generateRectangularMesh((width, height), (0,0), (1,1))
p = Problem(nodes, boundary_nodes, tris)
K = p.getStiffnessMatrix(lambda x,y: x*y)
self.assertTrue((sparse.triu(K, 1).T.toarray() == sparse.tril(K,-1).toarray()).all())
width = 5
height = 5
[nodes, boundary_nodes, tris] = generateRectangularMesh((width, height), (0,0), (1,1))
p = Problem(nodes, boundary_nodes, tris)
K = p.getStiffnessMatrix(lambda x,y: x*y)
self.assertTrue((sparse.triu(K, 1).T.toarray() == sparse.tril(K,-1).toarray()).all())
def testProblemStiffness(self):
width = 3
height = 3
[nodes, boundary_nodes, tris] = generateRectangularMesh((width, height), (0,0), (1,1))
p = Problem(nodes, boundary_nodes, tris)
stiffness = p.getStiffnessMatrix(lambda x,y: x*y).toarray()
self.assertEqual(stiffness[0,0], 34.0/9.0)
