def setUp(self): self.x = float(1.1) self.y = float(2.2) self.z = float(3.3) self.node_a = Node(1, self.x, self.y, self.z) self.node_b = Node(2, self.x + 1.0, self.y, self.z) self.node_c = Node(3, self.x, self.y + 1.0, self.z) self.config = ElmtConfig(ShapeType.Triangle, 1, False, False) self.comp_ID = 2 self.element = Element.Create(self.config, [self.node_a, self.node_b, self.node_c], [self.comp_ID])
def setUp(self): self.nodeset = NodeSet() self.nodeset_def_len = 10 for i in range(self.nodeset_def_len): id = i x = float(i) y = float(self.nodeset_def_len + i) z = float(self.nodeset_def_len**2 + i) n = Node(id, x, y, z) self.nodeset.add(n)
def _initialize_nodes(self): self.nodes = [] id_cnt = 0 for y in range(self.ny): tmp = [] for x in range(self.nx): tmp.append( Node(id_cnt, self.x_points[x], self.y_points[y], 0.0)) id_cnt += 1 self.nodes.append(tmp)
def testNodeSetAdd(self): msg = self.getCN() + "::testNodeSetAdd: " new_node = Node(len(self.nodeset) + 1, 1.0, 2.0, 3.0) self.nodeset.add(new_node) try: self.assertTrue(new_node in self.nodeset) msg += "PASS" print(msg) except: msg += "FAIL" print(msg) raise
def testNodeDistance(self): msg = self.getCN() + "::testNodeDistance: " node = Node(self.id + 1, self.x + 1, self.y + 2, self.z + 2) distance = self.node.Distance(node) expected_distance = 3.0 try: self.assertAlmostEqual(distance, expected_distance) msg += "PASS" print(msg) except: msg += "FAIL" print(msg) raise
def testMeshFieldAccess(self): msg = self.getCN() + "::testMeshFieldAccess: " self.nodeset = NodeSet() self.nodeset_def_len = 10 for i in range(self.nodeset_def_len): id = i x = float(i) y = float(self.nodeset_def_len + i) z = float(self.nodeset_def_len**2 + i) n = Node(id, x, y, z) self.nodeset.add(n) try: self.expDim = 2 self.spaceDim = 2 self.nummode = 7 self.verbose = False self.mesh.node = self.nodeset self.mesh.expDim = self.expDim self.mesh.spaceDim = self.spaceDim self.mesh.nummode = self.nummode self.mesh.verbose = self.verbose for node in self.nodeset: self.assertTrue(node in self.mesh.node) self.assertEqual(self.mesh.expDim, self.expDim) self.assertEqual(self.mesh.spaceDim, self.spaceDim) self.assertEqual(self.mesh.nummode, self.nummode) self.assertEqual(self.mesh.verbose, self.verbose) try: self.mesh.element = None except AttributeError: msg += "PASS" print(msg) except: msg += "FAIL" print(msg) raise
def setUp(self): self.id = int(1) self.x = float(1.1) self.y = float(2.2) self.z = float(3.3) self.node = Node(self.id, self.x, self.y, self.z)
class TestNode(unittest.TestCase): def getCN(self): return self.__class__.__name__ def setUp(self): self.id = int(1) self.x = float(1.1) self.y = float(2.2) self.z = float(3.3) self.node = Node(self.id, self.x, self.y, self.z) def testNodeConstructor(self): msg = self.getCN() + "::testNodeConstructor: " try: self.assertEqual(self.node.id, self.id) self.assertEqual(self.node.x, self.x) self.assertEqual(self.node.y, self.y) self.assertEqual(self.node.z, self.z) msg += "PASS" print(msg) except: msg += "FAIL" print(msg) raise def testNodeGetID(self): msg = self.__class__.__name__ + "::testNodeGetID: " try: self.assertEqual(self.node.GetID(), self.id) msg += "PASS" print(msg) except: msg += "FAIL" print(msg) raise def testNodeSetID(self): msg = self.getCN() + "::testNodeSetID: " self.id = 2 self.node.SetID(self.id) try: self.assertEqual(self.node.GetID(), self.id) msg += "PASS" print(msg) except: msg += "FAIL" print(msg) raise def testNodeDistance(self): msg = self.getCN() + "::testNodeDistance: " node = Node(self.id + 1, self.x + 1, self.y + 2, self.z + 2) distance = self.node.Distance(node) expected_distance = 3.0 try: self.assertAlmostEqual(distance, expected_distance) msg += "PASS" print(msg) except: msg += "FAIL" print(msg) raise def testNodeGetLoc(self): msg = self.getCN() + "::testNodeGetLoc: " expected_locations = [self.x, self.y, self.z] locations = self.node.GetLoc() try: for i in range(len(locations)): self.assertEqual(locations[i], expected_locations[i]) msg += "PASS" print(msg) except: msg += "FAIL" print(msg) raise def testNodeAbs2(self): msg = self.getCN() + "::testNodeAbs2: " expected_abs2 = self.x**2 + self.y**2 + self.z**2 abs2 = self.node.abs2() try: self.assertAlmostEqual(abs2, expected_abs2) msg += "PASS" print(msg) except: msg += "FAIL" print(msg) raise def testNodeFieldAccess(self): msg = self.getCN() + "::testNodeFieldAccess: " id = int(2) x = float(11.1) y = float(12.2) z = float(13.3) try: self.node.id = id self.node.x = x self.node.y = y self.node.z = z self.assertEqual(self.node.id, id) self.assertEqual(self.node.x, x) self.assertEqual(self.node.y, y) self.assertEqual(self.node.z, z) msg += "PASS" print(msg) except: msg += "FAIL" print(msg) raise