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)