Changeset 7873 for trunk/anuga_core/source/anuga/pmesh/test_mesh.py
- Timestamp:
- Jun 25, 2010, 11:10:58 AM (14 years ago)
- File:
-
- 1 edited
Legend:
- Unmodified
- Added
- Removed
-
trunk/anuga_core/source/anuga/pmesh/test_mesh.py
r7711 r7873 1194 1194 1195 1195 1196 def test_ungenerateFileLoading(self):1197 1198 fileName = tempfile.mktemp(".txt")1199 file = open(fileName,"w")1200 file.write(" 1 ?? ??\n\1201 0.0 0.0\n\1202 1.0 0.0\n\1203 1.0 1.0\n\1204 0.0 1.0\n\1205 0.0 0.0\n\1206 END\n\1207 2 ?? ??\n\1208 10.0 10.0\n\1209 10.0 20.0\n\1210 20.0 20.0\n\1211 10.0 10.0\n\1212 END\n\1213 END\n")1214 file.close()1215 1216 1217 a = Vertex (0.0, 0.0) #, attributes = [1.1])1218 b = Vertex (0.0, 40.0) #, attributes = [1.2])1219 c = Vertex (40.0,40.0) #, attributes = [1.3])1220 d = Vertex (40.0,0.0) #, attributes = [1.4])1221 1222 s1 = Segment(a,b)1223 s2 = Segment(b,c)1224 s3 = Segment(c,d)1225 s4 = Segment(d,a)1226 1227 m = Mesh(userVertices=[a,b,c,d], userSegments=[s1,s2,s3,s4])1228 dict = importUngenerateFile(fileName)1229 #os.remove(fileName)1230 1231 tag = "DSG"1232 Segment.set_default_tag(tag)1233 m.addVertsSegs(dict)1234 1235 # have to reset this , since it's a class attribute1236 Segment.set_default_tag("")1237 1238 self.failUnless(len(m.userSegments) ==11,1239 'Wrong segment list length.')1240 self.failUnless(len(m.userVertices) == 11,1241 'Wrong vertex list length.')1242 self.failUnless(m.userSegments[10].vertices[0] == m.userVertices[10],1243 'bad vertex on segment.')1244 self.failUnless(m.userSegments[10].vertices[1] == m.userVertices[8],1245 'Bad segment.')1246 self.failUnless(m.userSegments[10].tag == tag,1247 'wrong tag.')1248 1249 ## let's test the method1250 a = Vertex (0.0, 0.0) #, attributes = [1.1])1251 b = Vertex (0.0, 40.0) #, attributes = [1.2])1252 c = Vertex (40.0,40.0) #, attributes = [1.3])1253 d = Vertex (40.0,0.0) #, attributes = [1.4])1254 1255 s1 = Segment(a,b)1256 s2 = Segment(b,c)1257 s3 = Segment(c,d)1258 s4 = Segment(d,a)1259 1260 m = Mesh(userVertices=[a,b,c,d], userSegments=[s1,s2,s3,s4])1261 1262 tag = "DSG"1263 initial_tag = "PIG"1264 Segment.set_default_tag(initial_tag)1265 m.import_ungenerate_file(fileName, tag=tag)1266 1267 os.remove(fileName)1268 1269 self.failUnless(Segment.get_default_tag() == initial_tag,1270 'Wrong segment list length.')1271 1272 1273 # have to reset this , since it's a class attribute1274 Segment.set_default_tag("")1275 1276 self.failUnless(len(m.userSegments) ==11,1277 'Wrong segment list length.')1278 self.failUnless(len(m.userVertices) == 11,1279 'Wrong vertex list length.')1280 self.failUnless(m.userSegments[10].vertices[0] == m.userVertices[10],1281 'bad vertex on segment.')1282 self.failUnless(m.userSegments[10].vertices[1] == m.userVertices[8],1283 'Bad segment.')1284 self.failUnless(m.userSegments[10].tag == tag,1285 'wrong tag.')1286 1287 def test_import_ungenerate_file(self):1288 1289 fileName = tempfile.mktemp(".txt")1290 file = open(fileName,"w")1291 file.write(" 1 ?? ??\n\1292 10.0 10.0\n\1293 11.0 10.0\n\1294 11.0 11.0\n\1295 10.0 11.0\n\1296 10.0 10.0\n\1297 END\n\1298 2 ?? ??\n\1299 20.0 20.0\n\1300 20.0 30.0\n\1301 30.0 30.0\n\1302 END\n\1303 END\n")1304 file.close()1305 1306 1307 a = Vertex (0.0, 0.0) #, attributes = [1.1])1308 b = Vertex (0.0, 40.0) #, attributes = [1.2])1309 c = Vertex (40.0,40.0) #, attributes = [1.3])1310 d = Vertex (40.0,0.0) #, attributes = [1.4])1311 1312 s1 = Segment(a,b)1313 s2 = Segment(b,c)1314 s3 = Segment(c,d)1315 s4 = Segment(d,a)1316 1317 m = Mesh(userVertices=[a,b,c,d], userSegments=[s1,s2,s3,s4])1318 dict = importUngenerateFile(fileName)1319 #os.remove(fileName)1320 1321 tag = "DSG"1322 Segment.set_default_tag(tag)1323 m.addVertsSegs(dict)1324 1325 self.failUnless(len(m.userSegments) ==11,1326 'Wrong segment list length.')1327 self.failUnless(len(m.userVertices) == 11,1328 'Wrong vertex list length.')1329 1330 # Test the method1331 a = Vertex (0.0, 0.0) #, attributes = [1.1])1332 b = Vertex (0.0, 40.0) #, attributes = [1.2])1333 c = Vertex (40.0,40.0) #, attributes = [1.3])1334 d = Vertex (40.0,0.0) #, attributes = [1.4])1335 1336 s1 = Segment(a,b)1337 s2 = Segment(b,c)1338 s3 = Segment(c,d)1339 s4 = Segment(d,a)1340 1341 m = Mesh(userVertices=[a,b,c,d], userSegments=[s1,s2,s3,s4])1342 tag = "DSG"1343 initial_tag = "PIG"1344 Segment.set_default_tag(initial_tag)1345 m.import_ungenerate_file(fileName, tag=tag, region_tag="swamp")1346 1347 os.remove(fileName)1348 1349 self.failUnless(Segment.get_default_tag() == initial_tag,1350 'Wrong segment list length.')1351 m.export_mesh_file("swamp.tsh")1352 #print "m.userSegments",m.userSegments1353 self.failUnless(len(m.userSegments) ==11,1354 'Wrong segment list length.')1355 self.failUnless(len(m.userVertices) == 11,1356 'Wrong vertex list length.')1357 self.failUnless(len(m.regions) == 2,1358 'Wrong regions list length.')1359 self.failUnless(m.regions[0].getTag() == "swamp",1360 'Wrong regions tag.')1361 self.failUnless(m.regions[1].getTag() == "swamp",1362 'Wrong regions 1 tag.')1363 1364 # have to reset this , since it's a class attribute1365 Segment.set_default_tag("")1366 1367 1368 def test_import_ungenerate_file_different_region_tags(self):1369 1370 fileName = tempfile.mktemp(".txt")1371 file = open(fileName,"w")1372 file.write(" 1 ?? ??\n\1373 10.0 10.0\n\1374 11.0 10.0\n\1375 11.0 11.0\n\1376 10.0 11.0\n\1377 10.0 10.0\n\1378 END\n\1379 2 ?? ??\n\1380 20.0 20.0\n\1381 20.0 30.0\n\1382 30.0 30.0\n\1383 END\n\1384 END\n")1385 file.close()1386 1387 1388 a = Vertex (0.0, 0.0) #, attributes = [1.1])1389 b = Vertex (0.0, 40.0) #, attributes = [1.2])1390 c = Vertex (40.0,40.0) #, attributes = [1.3])1391 d = Vertex (40.0,0.0) #, attributes = [1.4])1392 1393 s1 = Segment(a,b)1394 s2 = Segment(b,c)1395 s3 = Segment(c,d)1396 s4 = Segment(d,a)1397 1398 m = Mesh(userVertices=[a,b,c,d], userSegments=[s1,s2,s3,s4])1399 dict = importUngenerateFile(fileName)1400 #os.remove(fileName)1401 1402 tag = "DSG"1403 Segment.set_default_tag(tag)1404 m.addVertsSegs(dict)1405 1406 self.failUnless(len(m.userSegments) ==11,1407 'Wrong segment list length.')1408 self.failUnless(len(m.userVertices) == 11,1409 'Wrong vertex list length.')1410 1411 # Test the method1412 a = Vertex (0.0, 0.0) #, attributes = [1.1])1413 b = Vertex (0.0, 40.0) #, attributes = [1.2])1414 c = Vertex (40.0,40.0) #, attributes = [1.3])1415 d = Vertex (40.0,0.0) #, attributes = [1.4])1416 1417 s1 = Segment(a,b)1418 s2 = Segment(b,c)1419 s3 = Segment(c,d)1420 s4 = Segment(d,a)1421 1422 m = Mesh(userVertices=[a,b,c,d], userSegments=[s1,s2,s3,s4])1423 tag = "DSG"1424 initial_tag = "PIG"1425 Segment.set_default_tag(initial_tag)1426 m.import_ungenerate_file(fileName, tag=tag, region_tag=["swamp","coastalp"])1427 1428 os.remove(fileName)1429 1430 self.failUnless(Segment.get_default_tag() == initial_tag,1431 'Wrong segment list length.')1432 m.export_mesh_file("swamp.tsh")1433 #print "m.userSegments",m.userSegments1434 self.failUnless(len(m.userSegments) ==11,1435 'Wrong segment list length.')1436 self.failUnless(len(m.userVertices) == 11,1437 'Wrong vertex list length.')1438 self.failUnless(len(m.regions) == 2,1439 'Wrong regions list length.')1440 self.failUnless(m.regions[0].getTag() == "swamp",1441 'Wrong regions tag.')1442 self.failUnless(m.regions[1].getTag() == "coastalp",1443 'Wrong regions 1 tag.')1444 1445 1446 # have to reset this , since it's a class attribute1447 Segment.set_default_tag("")1448 1449 1196 def test_addVertsSegs(self): 1450 1197 m = Mesh()
Note: See TracChangeset
for help on using the changeset viewer.