- Timestamp:
- Jan 13, 2009, 11:51:22 AM (15 years ago)
- File:
-
- 1 edited
Legend:
- Unmodified
- Added
- Removed
-
anuga_core/source/anuga/abstract_2d_finite_volumes/test_general_mesh.py
r5897 r6145 7 7 8 8 from anuga.config import epsilon 9 from Numeric import allclose, array, ones, Float10 9 from general_mesh import General_mesh 11 10 from anuga.coordinate_transforms.geo_reference import Geo_reference 12 11 12 import Numeric as num 13 13 14 14 … … 23 23 def test_get_vertex_coordinates(self): 24 24 from mesh_factory import rectangular 25 from Numeric import zeros, Float26 25 27 26 #Create basic mesh … … 30 29 31 30 32 assert allclose(domain.get_nodes(), nodes)31 assert num.allclose(domain.get_nodes(), nodes) 33 32 34 33 … … 41 40 for j in range(3): 42 41 k = triangles[i,j] #Index of vertex j in triangle i 43 assert allclose(V[3*i+j,:], nodes[k])42 assert num.allclose(V[3*i+j,:], nodes[k]) 44 43 45 44 def test_get_vertex_coordinates_with_geo_ref(self): … … 55 54 f = [4.0, 0.0] 56 55 57 nodes = array([a, b, c, d, e, f])56 nodes = num.array([a, b, c, d, e, f]) 58 57 59 58 nodes_absolute = geo.get_absolute(nodes) 60 59 61 60 #bac, bce, ecf, dbe, daf, dae 62 triangles = array([[1,0,2], [1,2,4], [4,2,5], [3,1,4]])61 triangles = num.array([[1,0,2], [1,2,4], [4,2,5], [3,1,4]]) 63 62 64 63 domain = General_mesh(nodes, triangles, 65 64 geo_reference = geo) 66 65 verts = domain.get_vertex_coordinates(triangle_id=0) 67 self.assert_( allclose(array([b,a,c]), verts))66 self.assert_(num.allclose(num.array([b,a,c]), verts)) 68 67 verts = domain.get_vertex_coordinates(triangle_id=0) 69 self.assert_( allclose(array([b,a,c]), verts))68 self.assert_(num.allclose(num.array([b,a,c]), verts)) 70 69 verts = domain.get_vertex_coordinates(triangle_id=0, 71 70 absolute=True) 72 self.assert_( allclose(array([nodes_absolute[1],71 self.assert_(num.allclose(num.array([nodes_absolute[1], 73 72 nodes_absolute[0], 74 73 nodes_absolute[2]]), verts)) 75 74 verts = domain.get_vertex_coordinates(triangle_id=0, 76 75 absolute=True) 77 self.assert_( allclose(array([nodes_absolute[1],76 self.assert_(num.allclose(num.array([nodes_absolute[1], 78 77 nodes_absolute[0], 79 78 nodes_absolute[2]]), verts)) … … 86 85 """ 87 86 from mesh_factory import rectangular 88 from Numeric import zeros, Float89 87 90 88 #Create basic mesh … … 93 91 94 92 95 assert allclose(domain.get_nodes(), nodes)93 assert num.allclose(domain.get_nodes(), nodes) 96 94 97 95 … … 104 102 for j in range(3): 105 103 k = triangles[i,j] #Index of vertex j in triangle i 106 assert allclose(V[j,:], nodes[k])104 assert num.allclose(V[j,:], nodes[k]) 107 105 108 106 … … 114 112 """ 115 113 from mesh_factory import rectangular 116 from Numeric import zeros, Float117 114 118 115 #Create basic mesh … … 121 118 122 119 value = [7] 123 assert allclose(domain.get_triangles(), triangles)124 assert allclose(domain.get_triangles([0,4]),120 assert num.allclose(domain.get_triangles(), triangles) 121 assert num.allclose(domain.get_triangles([0,4]), 125 122 [triangles[0], triangles[4]]) 126 123 … … 130 127 """ 131 128 from mesh_factory import rectangular 132 from Numeric import zeros, Float, array 133 134 a = [0.0, 0.0] 135 b = [0.0, 2.0] 136 c = [2.0, 0.0] 137 d = [0.0, 4.0] 138 e = [2.0, 2.0] 139 f = [4.0, 0.0] 140 141 nodes = array([a, b, c, d, e, f]) 129 130 a = [0.0, 0.0] 131 b = [0.0, 2.0] 132 c = [2.0, 0.0] 133 d = [0.0, 4.0] 134 e = [2.0, 2.0] 135 f = [4.0, 0.0] 136 137 nodes = num.array([a, b, c, d, e, f]) 142 138 #bac, bce, ecf, dbe, daf, dae 143 triangles = array([[1,0,2], [1,2,4], [4,2,5], [3,1,4]])139 triangles = num.array([[1,0,2], [1,2,4], [4,2,5], [3,1,4]]) 144 140 145 141 domain1 = General_mesh(nodes, triangles) … … 162 158 #print count 163 159 # 164 assert allclose(count, domain.number_of_triangles_per_node)160 assert num.allclose(count, domain.number_of_triangles_per_node) 165 161 166 162 # Check indices … … 196 192 f = [4.0, 0.0] 197 193 198 nodes = array([a, b, c, d, e, f])194 nodes = num.array([a, b, c, d, e, f]) 199 195 #bac, bce, ecf, dbe, daf, dae 200 triangles = array([[1,0,2], [1,2,4], [4,2,5], [3,1,4]])196 triangles = num.array([[1,0,2], [1,2,4], [4,2,5], [3,1,4]]) 201 197 202 198 domain = General_mesh(nodes, triangles) … … 204 200 # One node 205 201 L = domain.get_triangles_and_vertices_per_node(node=2) 206 assert allclose(L[0], [0, 2])207 assert allclose(L[1], [1, 1])208 assert allclose(L[2], [2, 1])202 assert num.allclose(L[0], [0, 2]) 203 assert num.allclose(L[1], [1, 1]) 204 assert num.allclose(L[2], [2, 1]) 209 205 210 206 # All nodes … … 213 209 for i, Lref in enumerate(ALL): 214 210 L = domain.get_triangles_and_vertices_per_node(node=i) 215 assert allclose(L, Lref)211 assert num.allclose(L, Lref) 216 212 217 213 … … 224 220 from mesh_factory import rectangular 225 221 from shallow_water import Domain 226 from Numeric import zeros, Float227 222 228 223 #Create basic mesh … … 239 234 from mesh_factory import rectangular 240 235 from shallow_water import Domain 241 from Numeric import zeros, Float242 236 243 237 #Create basic mesh … … 273 267 f = [4.0, 0.0] 274 268 275 nodes = array([a, b, c, d, e, f])269 nodes = num.array([a, b, c, d, e, f]) 276 270 277 271 nodes_absolute = geo.get_absolute(nodes) 278 272 279 273 #bac, bce, ecf, dbe, daf, dae 280 triangles = array([[1,0,2], [1,2,4], [4,2,5], [3,1,4]])274 triangles = num.array([[1,0,2], [1,2,4], [4,2,5], [3,1,4]]) 281 275 282 276 domain = General_mesh(nodes, triangles, … … 310 304 f = [4.0, 0.0] 311 305 312 nodes = array([a, b, c, d, e, f])306 nodes = num.array([a, b, c, d, e, f]) 313 307 314 308 nodes_absolute = geo.get_absolute(nodes) 315 309 316 310 # max index is 5, use 5, expect success 317 triangles = array([[1,5,2], [1,2,4], [4,2,5], [3,1,4]])311 triangles = num.array([[1,5,2], [1,2,4], [4,2,5], [3,1,4]]) 318 312 General_mesh(nodes, triangles, geo_reference=geo) 319 313 320 314 # max index is 5, use 6, expect assert failure 321 triangles = array([[1,6,2], [1,2,4], [4,2,5], [3,1,4]])315 triangles = num.array([[1,6,2], [1,2,4], [4,2,5], [3,1,4]]) 322 316 self.failUnlessRaises(AssertionError, General_mesh, 323 317 nodes, triangles, geo_reference=geo) 324 318 325 319 # max index is 5, use 10, expect assert failure 326 triangles = array([[1,10,2], [1,2,4], [4,2,5], [3,1,4]])320 triangles = num.array([[1,10,2], [1,2,4], [4,2,5], [3,1,4]]) 327 321 self.failUnlessRaises(AssertionError, General_mesh, 328 322 nodes, triangles, geo_reference=geo)
Note: See TracChangeset
for help on using the changeset viewer.