Context Navigation

source: inundation/ga/storm_surge/pmesh/test_mesh.py @ 1452

Last change on this file since 1452 was 1421, checked in by duncan, 20 years ago
speed up the drawing of alpha boundaries
File size: 58.2 KB

Rev	Line
[349]	1	#!/usr/bin/env python
	2	#
	3	import tempfile
	4	import unittest
	5	from mesh import *
	6	from load_mesh.loadASCII import *
[1023]	7	from coordinate_transforms.geo_reference import Geo_reference
[349]	8
	9	class meshTestCase(unittest.TestCase):
	10	def setUp(self):
	11	pass
	12
	13
	14
	15	def tearDown(self):
	16	pass
	17
	18	def testPointDistance(self):
	19	a = Point(0.0, 0.0)
	20	b = Point(0.0, 10.0)
	21
	22	self.failUnless( a.DistanceToPoint(b) == 10.0,
	23	'Point DistanceToPoint is wrong!')
	24
	25	def testVertexDistance(self):
	26	a = Vertex (0.0, 0.0)
	27	b = Vertex (0.0, 10.0)
	28
	29	self.failUnless( a.DistanceToPoint(b) == 10.0,
	30	'Point DistanceToPoint is wrong!')
	31
	32	def testTriangle(self):
	33	a = Vertex (0.0, 0.0)
	34	b = Vertex (0.0, 2.0)
	35	c = Vertex (2.0,0.0)
	36	d = Vertex (0.0, 4.0)
	37	e = Vertex (2.0, 2.0)
	38	f = Vertex (4.0,0.0)
	39
	40	t1 = Triangle(b,a,c)
	41	t2 = Triangle(b,c,e)
	42	t3 = Triangle(e,c,f)
	43	t4 = Triangle(d,b,e)
	44	t2.setNeighbors(t3,t4,t1)
	45
	46	self.failUnless( t2.neighbors[2].vertices[0] == b, 'Triangle initialisation is wrong!')
	47
	48
	49	def testSegment(self):
	50	a = Vertex (0.0, 0.0)
	51	b = Vertex (0.0, 10.0)
[969]	52	s = Segment(a,b, tag = 20)
[349]	53
	54	self.failUnless( s.vertices[0].DistanceToPoint(s.vertices[1]) == 10.0,
	55	'vertices in a segment are wrong')
	56
[969]	57	self.failUnless( s.tag == 20.0,
	58	'tag in a segment are wrong')
[349]	59
	60	def testdeleteUserVertex(self):
	61
	62
	63	mesh = Mesh()
	64	a = mesh.addUserVertex(0.0, 0.0)
	65	b = mesh.addUserVertex (0.0, 2.0)
	66	c = mesh.addUserVertex (2.0,0.0)
	67
	68	s1 = mesh.addUserSegment(a,b)
	69	s2 = mesh.addUserSegment(a,c)
	70	s3 = mesh.addUserSegment(c,b)
	71
	72	mesh.deleteMeshObject (a)
	73	self.failUnless(mesh.userSegments[0] == s3,
	74	'Bad segment. ')
	75	self.failUnless(len(mesh.userSegments) ==1,
	76	'Segments not deleted.')
	77	self.failUnless(len(mesh.userVertices) == 2,
	78	'Vertex not deleted.')
	79
	80
	81
	82	def testgenerateMesh(self):
	83	a = Vertex (0.0, 0.0)
	84	d = Vertex (0.0, 4.0)
	85	f = Vertex (4.0,0.0)
	86
	87	s1 = Segment(a,d)
	88	s2 = Segment(d,f)
	89	s3 = Segment(a,f)
	90
	91	r1 = Region(0.3, 0.3,tag = 1.3,maxArea = .6)
	92	#print r1
	93	m = Mesh(userVertices=[a,d,f], userSegments=[s1,s2,s3], regions=[r1] )
	94
	95	m.generateMesh("QApz", maxArea = 2.1 )
	96
	97	#print m
	98
	99	#m.plotMeshTriangle()
	100
	101	result = 1.414214
	102	delta = 0.00001
	103
	104	self.failUnless((m.meshTriangles[1].vertices[0].x < result + delta) or
	105	(m.meshTriangles[1].vertices[0].x > result - delta),
	106	'generated mesh is wrong!')
	107
	108	def test_regionalMaxArea(self):
	109	v0 = Vertex (0.0, 0.0)
	110	v1 = Vertex (6.0, 0.0)
	111	v2 = Vertex (6.0,6.0)
	112	v3 = Vertex (0.0,6.0)
	113
	114	s1 = Segment(v0,v1)
	115	s2 = Segment(v1,v2)
	116	s3 = Segment(v3,v2)
	117	s4 = Segment(v3,v0)
	118	s5 = Segment(v2,v0)
	119
	120	r1 = Region(3, 1,tag = 1.3)
	121	#print r1
	122	m = Mesh(userVertices=[v0,v1,v2,v3], userSegments=[s1,s2,s3,s4,s5], regions=[r1] )
	123
	124	m.generateMesh("Apz", maxArea = 36 )
	125
	126	#m.plotMeshTriangle()
	127	#print "len(m.meshTriangles)",len(m.meshTriangles)
	128
	129	self.failUnless(len(m.meshTriangles) == 2,
	130	'test_regionalMaxArea 1:generated mesh is wrong!')
	131
	132	## Another test case
	133	r1 = Region(3, 1,tag = 1.3)
	134	r2 = Region(1, 3,tag = 1.3, maxArea = 8)
	135	m = Mesh(userVertices=[v0,v1,v2,v3], userSegments=[s1,s2,s3,s4,s5],
	136	regions=[r1,r2] )
	137	m.generateMesh("Apz", maxArea = 36 )
	138	self.failUnless(len(m.meshTriangles) == 6,
	139	'test_regionalMaxArea 2:generated mesh is wrong!')
	140
	141	## Another test case
	142	r1 = Region(3, 1, tag = 1.3, maxArea = 8)
	143	r2 = Region(1, 3, tag = 1.3, maxArea = 8)
	144	m = Mesh(userVertices=[v0,v1,v2,v3], userSegments=[s1,s2,s3,s4,s5],
	145	regions=[r1,r2] )
	146	m.generateMesh("Apz", maxArea = 36 )
[1001]	147	#print "len(m.meshTriangles)",len(m.meshTriangles)
[349]	148	self.failUnless(len(m.meshTriangles) == 8,
	149	'test_regionalMaxArea 3:generated mesh is wrong!')
	150
	151	## Another test case
	152	r1 = Region(3, 1, tag = 1.3 )
	153	r2 = Region(1, 3, tag = 1.3, maxArea = 8)
	154	m = Mesh(userVertices=[v0,v1,v2,v3], userSegments=[s1,s2,s3,s4,s5],
	155	regions=[r1,r2] )
	156	m.generateMesh("Apz", maxArea = 8 )
	157	self.failUnless(len(m.meshTriangles) == 8,
	158	'test_regionalMaxArea 4:generated mesh is wrong!')
	159
	160	## Another test case
	161	r1 = Region(3, 1,tag = 1.3, maxArea = 8)
	162	r2 = Region(1, 3,tag = 1.3, maxArea = 8)
	163	m = Mesh(userVertices=[v0,v1,v2,v3], userSegments=[s1,s2,s3,s4,s5],
	164	regions=[r1,r2] )
	165	m.generateMesh("Apz", maxArea = 36,isRegionalMaxAreas = False )
	166	self.failUnless(len(m.meshTriangles) == 2,
	167	'test_regionalMaxArea 5:generated mesh is wrong!')
	168
	169	def testdeleteUserVertex(self):
	170	mesh = Mesh()
	171	a = mesh.addUserVertex(0.0, 0.0)
	172	b = mesh.addUserVertex (0.0, 2.0)
	173	c = mesh.addUserVertex (2.0,0.0)
	174
	175	s1 = mesh.addUserSegment(a,b)
	176	s2 = mesh.addUserSegment(a,c)
	177	s3 = mesh.addUserSegment(c,b)
	178
	179	mesh.deleteMeshObject (s2)
	180
	181	#print ",s2 in mesh.userSegments" ,s2 in mesh.userSegments
	182	self.failUnless(not(s2 in mesh.userSegments),
	183	'Bad segment. ')
	184	self.failUnless(len(mesh.userSegments) ==2,
	185	'Segments not deleted.')
	186	self.failUnless(len(mesh.userVertices) == 3,
	187	'Vertex deleted, instead of segment.')
	188
	189	def testTriangleArea(self):
	190	a = Vertex (10.0, 10.0)
	191	b = Vertex (10.0, 20.0)
	192	c = Vertex (20.0,10.0)
	193
	194	d = Vertex (-20.0, 0.0)
	195	e = Vertex (-20.0, -20.0)
	196	f = Vertex (20.0,-20.0)
	197
	198	t1 = Triangle(b,a,c)
	199	t2 = Triangle(e,d,f)
	200
	201	# print "t1", t1
	202	# print "t1 area ", t1.calcArea()
	203	# print "t2", t2
	204	# print "t2 area ", t2.calcArea()
	205	self.failUnless( t1.calcArea() == 50 and t2.calcArea() == 400, 'Triangle area is wrong!')
	206	def testisUserSegmentNew (self):
	207	mesh = Mesh()
	208	a = mesh.addUserVertex(0.0, 0.0)
	209	b = mesh.addUserVertex (0.0, 2.0)
	210	c = mesh.addUserVertex (2.0,0.0)
	211	d = mesh.addUserVertex (2.0,3.0)
	212
	213	s1 = mesh.addUserSegment(a,b)
	214	s2 = mesh.addUserSegment(a,c)
	215	s3 = mesh.addUserSegment(c,b)
	216
	217	self.failUnless(mesh.isUserSegmentNew(a,d) ,
	218	'Segment should be new. ')
	219	self.failUnless(not(mesh.isUserSegmentNew(a,b)) ,
	220	'Segment should not be new. ')
	221
[702]	222
	223	def testisUserSegmentNew (self):
	224	mesh = Mesh()
	225	a = mesh.addUserVertex(0.0, 0.0)
	226	b = mesh.addUserVertex (0.0, 2.0)
	227	c = mesh.addUserVertex (2.0,0.0)
	228	d = mesh.addUserVertex (2.0,3.0)
[349]	229
[702]	230	s1 = mesh.addUserSegment(a,b)
	231	s2 = mesh.addUserSegment(a,c)
	232	s3 = mesh.addUserSegment(c,b)
[349]	233
[702]	234	self.failUnless(mesh.representedUserSegment(a,d) == None,
	235	'Segment should be new. ')
	236	self.failUnless(mesh.representedUserSegment(a,b) == s1 ,
	237	'Segment should not be new. ')
[349]	238
[1401]	239	def testautoSegment(self):
[702]	240	p0 = Vertex (0.0, 0.0)
	241	p1 = Vertex (0.0, 4.0)
	242	p2 = Vertex (4.0,4.0)
	243	p3 = Vertex (4.0,0.0)
[349]	244
[702]	245	s1 = Segment(p0,p1)
[349]	246
[702]	247	m = Mesh(userVertices=[p0, p1, p2, p3], userSegments=[s1] )
[349]	248	m.autoSegment()
	249
	250	#print 'Len', len(m.userSegments)
[702]	251	self.failUnless(len(m.getUserSegments()) == 4 ,
	252	'userSegments is wrong!')
	253
	254	m.autoSegment()
	255	self.failUnless(len(m.getUserSegments()) == 4 ,
	256	'userSegments is wrong!')
	257
[1401]	258	def testautoSegmentII(self):
[702]	259	p1 = Vertex (3.0, 4.0)
	260	p2 = Vertex (3.0,2.0)
	261	p3 = Vertex (3.0,0.0)
	262	p4 = Vertex (6.0, 4.0)
	263	p5 = Vertex (6.0,2.0)
	264	p0 = Vertex (6.0,0.0)
[349]	265
[702]	266
	267	s1 = Segment(p2,p3)
	268	s2 = Segment(p4,p5)
	269
	270	m = Mesh(userVertices=[p0, p1, p2, p3, p4, p5],
	271	userSegments=[s1, s2])
	272
	273	m.autoSegment()
	274
	275	s3 = m.representedAlphaUserSegment(p3,p0)
	276	self.failUnless(not (s3 == None) ,
[349]	277	'userSegments is wrong!')
[702]	278
	279
	280	s6 = m.representedAlphaUserSegment(p1,p4)
	281	self.failUnless(not (s6 == None) ,
	282	'userSegments is wrong!')
	283
	284	# remove a segment, add a point, autosegment
	285	m.alphaUserSegments.remove(s3)
	286	p6 = Vertex (1.0, 2.0)
	287	m.userVertices.append(p6)
[1421]	288
[702]	289	m.autoSegment()
	290
	291	s1_now = m.representedUserSegment(p3,p2)
	292	self.failUnless(s1_now == s1 ,
	293	'userSegments is wrong!')
	294
	295	s2_now = m.representedUserSegment(p5,p4)
	296	self.failUnless(s2_now == s2 ,
	297	'userSegments is wrong!')
	298
	299	s3 = m.representedAlphaUserSegment(p3,p6)
	300	self.failUnless(not (s3 == None) ,
	301	'userSegments is wrong!')
	302
	303	s4 = m.representedAlphaUserSegment(p3,p6)
	304	self.failUnless(not (s4 == None) ,
	305	'userSegments is wrong!')
	306
	307	s5 = m.representedAlphaUserSegment(p4,p6)
	308	self.failUnless(s5 == None ,
	309	'userSegments is wrong!')
	310	#print m
	311
[349]	312	def testRegions(self):
	313	a = Vertex (0.0, 0.0)
	314	b = Vertex (0.0, 2.0)
	315	c = Vertex (2.0,0.0)
	316	d = Vertex (0.0, 4.0)
	317	e = Vertex (2.0, 2.0)
	318	f = Vertex (4.0,0.0)
	319	g = Vertex (0.0,-2.0)
	320
	321	Segment.set_default_tag("")
	322	s1 = Segment(a,b)
	323	s2 = Segment(b,e)
	324	s3 = Segment(e,c)
	325	s4 = Segment(c,a)
	326	s5 = Segment(b,d)
	327	s6 = Segment(e,d)
	328	s7 = Segment(e,f)
	329	s8 = Segment(c,f)
	330	s9 = Segment(g,c)
	331	s10 = Segment(g,a)
	332
	333	r1 = Region(0.1,0.1,tag="22")
	334	r2 = Region(0.1,2.1,tag="11")
	335	r3 = Region(2.1,0.1)
	336
	337	m = Mesh(userVertices=[a,b,c,d,e,f,g], userSegments=[s1,s2,s3,s4,s5,s6,s7,s8,s9,s10], regions=[r1,r2,r3] )
	338	m.generateMesh("QApz", maxArea = 2.1 )
	339	#print m
	340	Triangulation = m.getTriangulation()
	341	#print Triangulation[0].attribute
	342	#print Triangulation[1].attribute
	343	#print Triangulation[2].attribute
	344	#print Triangulation[3].attribute
	345	#print Triangulation[4].attribute
	346
	347	self.failUnless(Triangulation[0].attribute == "" and
	348	Triangulation[1].attribute == "22" and
	349	Triangulation[2].attribute == "" and
	350	Triangulation[3].attribute == "11" and
	351	Triangulation[4].attribute == "22" ,
	352	'region attributes are wrong!')
	353
	354	def test_vertexAttribs(self):
	355	a = Vertex (0.0, 0.0, attributes = [12.0,2.0])
	356	d = Vertex (0.0, 4.0, attributes = [9.0,7.0])
	357	f = Vertex (4.0,0.0, attributes = [14.0,3.0])
	358
	359	Segment.set_default_tag("")
	360	s1 = Segment(a,d)
	361	s2 = Segment(d,f)
	362	s3 = Segment(a,f)
	363
	364	r1 = Region(0.3, 0.3, tag = 88.9)
	365
	366	m = Mesh(userVertices=[a,d,f], userSegments=[s1,s2,s3], regions=[r1])
	367
	368	m.generateMesh("QApz", maxArea = 2.1)
	369
	370	vert = m.getMeshVertices()
	371
	372	self.failUnless(vert[0].attributes == [12.0, 2.0] and
	373	vert[1].attributes == [9.0, 7.0] and
	374	vert[2].attributes == [14.0,3.0] and
	375	vert[3].attributes == [12.232233047033631, 4.4142135623730949] and
	376	vert[4].attributes == [13.0, 2.5] ,
	377	'vertex attributes are wrong!')
	378
	379
	380	def test_vertexAttribs2(self):
	381
	382	a = Vertex (0.0, 0.0)
	383	d = Vertex (0.0, 4.0)
	384	f = Vertex (4.0,0.0)
	385
	386	Segment.set_default_tag("")
	387	s1 = Segment(a,d)
	388	s2 = Segment(d,f)
	389	s3 = Segment(a,f)
	390
	391	r1 = Region(0.3, 0.3, tag = 88.9)
	392
	393	m = Mesh(userVertices=[a,d,f], userSegments=[s1,s2,s3], regions=[r1])
	394
	395	m.generateMesh("QApz", maxArea = 2.1 )
	396
	397	vert = m.getMeshVertices()
	398	self.failUnless(vert[0].attributes == [] and
	399	vert[1].attributes == [] and
	400	vert[2].attributes == [] and
	401	vert[3].attributes == [] and
	402	vert[4].attributes == [],
	403	'vertex attributes are wrong!')
	404
[969]	405	def test_segtag(self):
[349]	406
	407	a = Vertex (0.0, 0.0)
	408	d = Vertex (0.0, 4.0)
	409	f = Vertex (4.0,0.0)
	410
[969]	411	s1 = Segment(a,d,tag = 5)
	412	s2 = Segment(d,f,tag = 7)
	413	s3 = Segment(a,f,tag = 9)
[349]	414
	415	m = Mesh(userVertices=[a,d,f], userSegments=[s1,s2,s3])
	416
	417	m.generateMesh("QApz", maxArea = 2.1 )
	418
	419	seg = m.getMeshSegments()
	420
[969]	421	#print "seg[0].tag"
	422	#print seg[0].tag
	423	#print "seg[0].tag"
	424	self.failUnless(seg[0].tag == 5 and
	425	seg[1].tag == 7 and
	426	seg[2].tag == 9 and
	427	seg[3].tag == 7 and
	428	seg[4].tag == 9 and
	429	seg[5].tag == 7 and
	430	seg[6].tag == 5,
	431	'seg tags are wrong')
[349]	432
[969]	433	def test_segtag2(self):
[349]	434
	435	a = Vertex (0.0, 0.0)
	436	d = Vertex (0.0, 4.0)
	437	f = Vertex (4.0,0.0)
	438	e = Vertex (1.0,1.0)
	439
	440	s1 = Segment(a,d)
	441	s2 = Segment(d,f)
	442	s3 = Segment(a,f)
	443	s4 = Segment(a,e)
	444
	445	m = Mesh(userVertices=[a,d,f,e], userSegments=[s1,s2,s3,s4])
	446
	447	m.generateMesh("QApz", maxArea = 2.1)
	448
	449	seg = m.getMeshSegments()
[969]	450	self.failUnless(seg[0].tag == "exterior" and
	451	seg[1].tag == "exterior" and
	452	seg[2].tag == "exterior" and
	453	seg[3].tag == "" and
	454	seg[4].tag == "exterior",
	455	'2nd seg tags are wrong')
[349]	456
	457	def test_asciiFile(self):
	458
[364]	459	a = Vertex (0.0, 0.0) #, attributes = [1.1])
	460	d = Vertex (0.0, 4.0) #, attributes = [1.2])
	461	f = Vertex (4.0,0.0) #, attributes = [1.3])
	462	e = Vertex (1.0,1.0) #, attributes = [1.4])
[349]	463
	464	s1 = Segment(a,d)
	465	s2 = Segment(d,f)
	466	s3 = Segment(a,f)
	467	s4 = Segment(a,e)
	468
	469	m = Mesh(userVertices=[a,d,f,e], userSegments=[s1,s2,s3,s4])
	470
	471	m.generateMesh("QApz", maxArea = 2.1 )
	472
	473	seg = m.getMeshSegments()
	474
[996]	475	fileName = tempfile.mktemp(".tsh")
[1001]	476	m.export_mesh_file(fileName)
[349]	477	file = open(fileName)
	478	lFile = file.read().split('\n')
	479	file.close()
	480	os.remove(fileName)
	481
	482	#print "@^@^"
	483	#for l in lFile:
	484	# print l,"<"
	485	#print "@^@^"
[364]	486
[721]	487	# no need to check the title again
	488	#self.failUnless(lFile[0] == "5 0 # <vertex #> <x> <y> [attributes] ...Triangulation Vertices..."
	489	# ,'Ascii file is wrong, vertex title')
[364]	490	self.failUnless(lFile[1] == "0 0.0 0.0 " and #1.1 " and
	491	lFile[2] == "1 0.0 4.0 " and #1.2 " and
	492	lFile[3] == "2 4.0 0.0 " and #1.3 " and
	493	lFile[4] == "3 1.0 1.0 " and #1.4 " and
	494	lFile[5] == "4 2.0 2.0 " #1.25 "
[349]	495	,
	496	'Ascii file is wrong, vertex')
[364]	497
[721]	498	#self.failUnless(lFile[6] == "# attribute column titles ...Triangulation Vertex Titles..."
	499	# ,'Ascii file is wrong, attribute column title')
	500	self.failUnless(lFile[8] == "0 3 2 4 -1 2 3 " and
[364]	501	lFile[9] == "1 1 0 3 3 2 -1 " and
	502	lFile[10] == "2 3 4 1 -1 1 0 " and
	503	lFile[11] == "3 2 3 0 1 -1 0 "
	504	,
[349]	505	'Ascii file is wrong, triangle')
	506
[721]	507	self.failUnless( lFile[13] == "0 0 1 exterior" and
[714]	508	lFile[14] == "1 1 4 exterior" and
	509	lFile[15] == "2 2 0 exterior" and
[650]	510	lFile[16] == "3 0 3 " and
[714]	511	lFile[17] == "4 4 2 exterior" ,
[349]	512	'Ascii file is wrong, segment')
	513
[721]	514	# self.failUnless(lFile[18] == '4 0 # <vertex #> <x> <y> [attributes] ...Mesh Vertices...',
	515	# 'Ascii file is wrong, Mesh Vertices Title')
[349]	516
[364]	517	self.failUnless(lFile[19] == '0 0.0 0.0 ' and #1.1 ' and
	518	lFile[20] == '1 0.0 4.0 ' and #1.2 ' and
	519	lFile[21] == '2 4.0 0.0 ' and #1.3 ' and
	520	lFile[22] == '3 1.0 1.0 ' #1.4 '
	521	,
[349]	522	'Ascii file is wrong, Mesh Vertices II')
	523
[721]	524	self.failUnless(lFile[24] == '0 0 1 ' and
[364]	525	lFile[25] == '1 1 2 ' and
	526	lFile[26] == '2 0 2 ' and
[721]	527	lFile[27] == '3 0 3 '
	528	,'Ascii file is wrong, Mesh Segments')
[364]	529
[349]	530
	531	def test_ascii_file(self):
	532
[364]	533	a = Vertex (0.0, 0.0) #, attributes = [1.1])
	534	d = Vertex (0.0, 4.0) #, attributes = [1.2])
	535	f = Vertex (4.0,0.0) #, attributes = [1.3])
	536	e = Vertex (1.0,1.0) #, attributes = [1.4])
[349]	537
	538	s1 = Segment(a,d)
	539	s2 = Segment(d,f)
	540	s3 = Segment(a,f)
	541	s4 = Segment(a,e)
	542
	543	m = Mesh(userVertices=[a,d,f,e], userSegments=[s1,s2,s3,s4])
	544
	545	m.generateMesh("QApz", maxArea = 2.1 )
	546
	547	seg = m.getMeshSegments()
	548
[996]	549	fileName = tempfile.mktemp(".tsh")
[1001]	550	m.export_mesh_file(fileName)
[349]	551	file = open(fileName)
	552	lFile = file.read().split('\n')
	553	file.close()
	554	os.remove(fileName)
	555
[852]	556	#print "@^@^"
	557	#for l in lFile:
	558	# print l,"<"
	559	#print "@^@^"
[721]	560	self.failUnless(lFile[0] == "5 0 # <# of verts> <# of vert attributes>, next lines <vertex #> <x> <y> [attributes] ...Triangulation Vertices..."
[349]	561	,
	562	'Ascii file is wrong, vertex title')
[364]	563	self.failUnless(lFile[1] == "0 0.0 0.0 " and #1.1 " and
	564	lFile[2] == "1 0.0 4.0 " and #1.2 " and
	565	lFile[3] == "2 4.0 0.0 " and #1.3 " and
	566	lFile[4] == "3 1.0 1.0 " and #1.4 " and
	567	lFile[5] == "4 2.0 2.0 " #1.25 "
[349]	568	,
	569	'Ascii file is wrong, vertex')
[364]	570
	571	self.failUnless(lFile[6] == "# attribute column titles ...Triangulation Vertex Titles..."
[349]	572	,
[364]	573	'Ascii file is wrong, attribute column title')
[721]	574	self.failUnless(lFile[7] == "4 # <# of triangles>, next lines <triangle #> [<vertex #>] [<neigbouring triangle #>] [attribute of region] ...Triangulation Triangles..." and
[364]	575	lFile[8] == "0 3 2 4 -1 2 3 " and
	576	lFile[9] == "1 1 0 3 3 2 -1 " and
	577	lFile[10] == "2 3 4 1 -1 1 0 " and
	578	lFile[11] == "3 2 3 0 1 -1 0 "
	579	,
[349]	580	'Ascii file is wrong, triangle')
	581
[969]	582	self.failUnless(lFile[12] == "5 # <# of segments>, next lines <segment #> <vertex #> <vertex #> [boundary tag] ...Triangulation Segments..." and
[714]	583	lFile[13] == "0 0 1 exterior" and
	584	lFile[14] == "1 1 4 exterior" and
	585	lFile[15] == "2 2 0 exterior" and
[650]	586	lFile[16] == "3 0 3 " and
[714]	587	lFile[17] == "4 4 2 exterior" ,
[349]	588	'Ascii file is wrong, segment')
	589
[721]	590	self.failUnless(lFile[18] == '4 0 # <# of verts> <# of vert attributes>, next lines <vertex #> <x> <y> [attributes] ...Mesh Vertices...',
[349]	591	'Ascii file is wrong, Mesh Vertices Title')
	592
[364]	593	self.failUnless(lFile[19] == '0 0.0 0.0 ' and #1.1 ' and
	594	lFile[20] == '1 0.0 4.0 ' and #1.2 ' and
	595	lFile[21] == '2 4.0 0.0 ' and #1.3 ' and
	596	lFile[22] == '3 1.0 1.0 ' #1.4 '
	597	,
[349]	598	'Ascii file is wrong, Mesh Vertices II')
	599
[969]	600	self.failUnless(lFile[23] == '4 # <# of segments>, next lines <segment #> <vertex #> <vertex #> [boundary tag] ...Mesh Segments...' and
[364]	601	lFile[24] == '0 0 1 ' and
	602	lFile[25] == '1 1 2 ' and
	603	lFile[26] == '2 0 2 ' and
	604	lFile[27] == '3 0 3 ' and
[721]	605	lFile[28] == '0 # <# of holes>, next lines <Hole #> <x> <y> ...Mesh Holes...' and
	606	lFile[29] == '0 # <# of regions>, next lines <Region #> <x> <y> <tag>...Mesh Regions...'
[349]	607	,
	608	'Ascii file is wrong, Mesh Segments')
	609
	610
	611	def test_thinoutVertices(self):
	612
	613	v1 = Vertex(-20,-20)
	614	v2 = Vertex(-11,-11)
	615	v3 = Vertex(-10,-10)
	616	v4 = Vertex(-9,-1)
	617	v5 = Vertex(-8,2)
	618	v6 = Vertex(6,3)
	619	v7 = Vertex(12,9)
	620	v8 = Vertex(15,3)
	621	v9 = Vertex(24,3)
	622	m = Mesh(userVertices = [v1,v2,v3,v4,v5,v6,v7,v8,v9])
	623	m.thinoutVertices(10)
	624
	625	self.failUnless(v1 in m.userVertices,
	626	'test_thinoutVertices, test 1 failed')
	627	self.failUnless(v3 in m.userVertices,
	628	'test_thinoutVertices, test 2 failed')
	629	self.failUnless(v4 in m.userVertices,
	630	'test_thinoutVertices, test 3 failed')
	631	self.failUnless(v6 in m.userVertices,
	632	'test_thinoutVertices, test 4 failed')
	633	self.failUnless(v7 in m.userVertices,
	634	'test_thinoutVertices, test 5 failed')
	635	self.failUnless(v9 in m.userVertices,
	636	'test_thinoutVertices, test 6 failed')
	637	self.failUnless(v5 not in m.userVertices,
	638	'test_thinoutVertices, test 7 failed')
	639	self.failUnless(v2 not in m.userVertices,
	640	'test_thinoutVertices, test 8 failed')
	641	self.failUnless(v8 not in m.userVertices,
	642	'test_thinoutVertices, test 9 failed')
	643
	644	def test_same_x_y(self):
	645	v = Point(7,8)
	646	f = Point(7,8)
	647	f.same_x_y(v)
	648
	649	self.failUnless(f.same_x_y(v),
	650	'same_x_y True failed')
	651	e = Point(7,9)
	652	self.failUnless(not f.same_x_y(e),
	653	'same_x_y False failed')
	654
[1023]	655	def test_import_tsh(self):
[349]	656
	657	a_att = [5.0,2.0]
	658	d_att =[4.0,2.0]
	659	f_att = [3.0,2.0]
	660	e_att = [2.0,2.0]
	661	a_xy = [0.0, 0.0]
[364]	662	a = Vertex ( a_xy[0],a_xy[1]) #, attributes =a_att)
	663	d = Vertex (0.0, 4.0) #, attributes =d_att)
	664	f = Vertex (4.0,0.0) #, attributes =f_att)
	665	e = Vertex (1.0,1.0) #, attributes =e_att)
[349]	666
[969]	667	s1 = Segment(a,d, tag = "50")
	668	s2 = Segment(d,f, tag = "40")
	669	s3 = Segment(a,f, tag = "30")
	670	s4 = Segment(a,e, tag = "20")
[349]	671
	672	r1 = Region(0.3, 0.3,tag = "1.3")
[1023]	673	geo = Geo_reference(8.9,8.9,65)
[349]	674	m = Mesh(userVertices=[a,d,f,e],
	675	userSegments=[s1,s2,s3,s4],
[1023]	676	regions=[r1],
	677	geo_reference=geo)
[349]	678
	679	m.generateMesh("QApz", maxArea = 2.1)
	680	fileName = tempfile.mktemp(".tsh")
	681	#print "dgs!!!"
	682	#print "****************** fileName", fileName
[1001]	683	m.export_mesh_file(fileName)
[406]	684	#print "******************"
	685	#print "m", m
	686	#print "******************"
[349]	687	m_returned = importMeshFromFile(fileName)
[406]	688	#print "m_returned",m_returned
	689	#print "******************"
[349]	690	#print "****************** fileName", fileName
	691	os.remove(fileName)
	692	self.failUnless(0 == m.__cmp__(m_returned),
	693	'loading and saving of a mesh failed')
[1023]	694	# do this when .msh supports geo_refs
	695	#self.failUnless(m.geo_reference == m_returned.geo_reference,
	696	# 'loading and saving of a mesh geo refs failed')
[349]	697
[1023]	698	def test_import_mesh(self):
[349]	699
[1023]	700	a_att = [5.0,2.0]
	701	d_att =[4.0,2.0]
	702	f_att = [3.0,2.0]
	703	e_att = [2.0,2.0]
	704	a_xy = [0.0, 0.0]
	705	a = Vertex ( a_xy[0],a_xy[1]) #, attributes =a_att)
	706	d = Vertex (0.0, 4.0) #, attributes =d_att)
	707	f = Vertex (4.0,0.0) #, attributes =f_att)
	708	e = Vertex (1.0,1.0) #, attributes =e_att)
	709
	710	s1 = Segment(a,d, tag = "50")
	711	s2 = Segment(d,f, tag = "40")
	712	s3 = Segment(a,f, tag = "30")
	713	s4 = Segment(a,e, tag = "20")
	714
	715	r1 = Region(0.3, 0.3,tag = "1.3")
[1073]	716	geo = Geo_reference(65,8.9,8.9)
[1023]	717	m = Mesh(userVertices=[a,d,f,e],
	718	userSegments=[s1,s2,s3,s4],
	719	regions=[r1],
	720	geo_reference=geo)
	721
	722	m.generateMesh("QApz", maxArea = 2.1)
	723	fileName = tempfile.mktemp(".msh")
	724	#print "dgs!!!"
	725	#print "****************** fileName", fileName
	726	m.export_mesh_file(fileName)
	727	#print "******************"
	728	#print "m", m
	729	#print "******************"
	730	m_returned = importMeshFromFile(fileName)
	731	#print "m_returned",m_returned
	732	#print "******************"
	733	#print "****************** fileName", fileName
	734	os.remove(fileName)
[1034]	735	#print "m.geo_reference",m.geo_reference
	736	#print "m_returned.geo_reference,",m_returned.geo_reference
[1023]	737	self.failUnless(0 == m.__cmp__(m_returned),
	738	'loading and saving of a mesh failed')
	739	self.failUnless(m.geo_reference == m_returned.geo_reference,
	740	'loading and saving of a mesh geo refs failed')
	741
[349]	742	def test_normaliseMesh(self):
	743
	744	a_att = [5.0,2.0]
	745	d_att =[4.0,2.0]
	746	f_att = [3.0,2.0]
	747	e_att = [2.0,2.0]
	748	a_xy = [10.0, 10.0]
	749	a = Vertex ( a_xy[0],a_xy[1], attributes =a_att)
	750	d = Vertex (15.0, 10.0, attributes =d_att)
	751	f = Vertex (10.0,20.0, attributes =f_att)
	752	e = Vertex (15.0,20.0, attributes =e_att)
	753
[969]	754	s1 = Segment(a,d, tag = 50)
	755	s2 = Segment(d,e, tag = 40)
	756	s3 = Segment(e,f, tag = 30)
	757	s4 = Segment(f,a, tag = 20)
[349]	758
	759	r1 = Region(0.3, 0.3,tag = 1.3)
	760	m = Mesh(userVertices=[a,d,f,e],
	761	userSegments=[s1,s2,s3,s4],
	762	regions=[r1])
	763	m.normaliseMesh(1,0,1)
	764	[xmin, ymin, xmax, ymax] = m.boxsize()
	765	[attmin, attmax] = m.maxMinVertAtt(0)
	766	self.failUnless(attmin == 0.0 and attmax == 1.0,
	767	'normalise failed')
	768	self.failUnless(xmin == 0.0 and ymin == 0.0 and xmax == 0.5 and ymax == 1.0,
	769	'normalise failed')
	770	m.normaliseMesh(200,-100,5)
	771	[xmin, ymin, xmax, ymax] = m.boxsize()
	772	[attmin, attmax] = m.maxMinVertAtt(0)
	773	self.failUnless(attmin == 0.0 and attmax == 5.0,
	774	'normalise failed')
	775	self.failUnless(xmin == -100.0 and ymin == -100.0 and xmax == 0.0 and ymax == 100.0,
	776	'normalise failed')
[1401]	777
	778	def test_exportASCIIsegmentoutlinefile(self):
[349]	779	a = Vertex (0,0)
	780	b = Vertex (0,3)
	781	c = Vertex (3,3)
	782	d = Vertex (1,2)
	783	e = Vertex (3,1)
	784
[969]	785	s1 = Segment(a,b, tag = "50")
	786	s2 = Segment(b,c, tag = "40")
	787	s3 = Segment(c,a, tag = "30")
[349]	788
	789	r1 = Region(2, 1,tag = "1.3")
	790	h1 = Hole(1,4)
	791	m = Mesh(userVertices=[a,b,c,d,e],
	792	userSegments=[s1,s2,s3],
	793	regions=[r1],
	794	holes = [h1])
	795
	796	m.generateMesh("QApz", maxArea = 2.1)
	797	#print "mesh **************dsg", m
	798
	799	fileName = tempfile.mktemp(".tsh")
	800	m.exportASCIIsegmentoutlinefile(fileName)
	801
	802	m_returned = importMeshFromFile(fileName)
	803
[1001]	804	#print "m_returned ****",m_returned
[349]	805	#print "****************** fileName", fileName
	806	os.remove(fileName)
	807
	808	#Trim mesh, so it should like like m_returned
	809	m.meshVertices = []
	810	m.meshTriangles = []
	811	m.meshSegments = []
	812	m.userVertices=[a,b,c]
[1001]	813	#print "mesh **************dsg", m
[349]	814	#print "(m.__cmp__(m_returned)", m.__cmp__(m_returned)
	815	self.failUnless(0 == m.__cmp__(m),
	816	'test_exportASCIIsegmentoutlinefile:loading and saving of a mesh failed')
	817	self.failUnless(0 == m.__cmp__(m_returned),
	818	'test_exportASCIIsegmentoutlinefile:loading and saving of a mesh failed')
[1401]	819
	820	def test_exportASCIIsegmentoutlinefile2(self):
	821	a = Vertex (0,0)
	822	b = Vertex (0,1)
	823	c = Vertex (1,0)
	824	d = Vertex (1,1)
	825	e = Vertex (0.5,0.5)
	826	f = Vertex (0.6,0.6)
	827
	828	s1 = Segment(a,e, tag = "50")
	829	s2 = Segment(b,e, tag = "40")
	830	s3 = Segment(c,e, tag = "30")
	831	s4 = Segment(d,e, tag = "30")
	832
	833	r1 = Region(2, 1,tag = "1.3")
	834	h1 = Hole(1,4)
	835	m = Mesh(userVertices=[a,b,c,d,e],
	836	userSegments=[s1,s2,s3,s4],
	837	regions=[r1],
	838	holes = [h1])
	839
	840	fileName = tempfile.mktemp(".tsh")
	841	m.exportASCIIsegmentoutlinefile(fileName)
	842
	843	m_returned = importMeshFromFile(fileName)
	844	#print "****************** fileName", fileName
	845	os.remove(fileName)
[349]	846
[1401]	847	#Trim mesh, so it should like like m_returned
	848	m.meshVertices = []
	849	m.meshTriangles = []
	850	m.meshSegments = []
	851	m.userVertices=[a,e,d,b,c]
	852	#print "mesh **************dsg", m
	853	#print "(m.__cmp__(m_returned)", m.__cmp__(m_returned)
	854	self.failUnless(0 == m.__cmp__(m),
	855	'test_exportASCIIsegmentoutlinefile:loading and saving of a mesh failed')
	856	self.failUnless(0 == m.__cmp__(m_returned),
	857	'test_exportASCIIsegmentoutlinefile:loading and saving of a mesh failed')
	858
[852]	859	def test_loadxy(self):
	860	"""
	861	To test the mesh side of loading xya files.
	862	Not the loading of xya files
	863	"""
	864	import os
	865	import tempfile
	866
	867	fileName = tempfile.mktemp(".xya")
	868	file = open(fileName,"w")
	869	file.write("elevation speed \n\
	870	1.0 0.0 10.0 0.0\n\
	871	0.0 1.0 0.0 10.0\n\
	872	1.0 0.0 10.4 40.0\n")
	873	file.close()
	874	#print fileName
	875	m = importMeshFromFile(fileName)
	876	os.remove(fileName)
	877	self.failUnless(m.userVertices[0].x == 1.0,
	878	'loadxy, test 1 failed')
	879	self.failUnless(m.userVertices[0].y == 0.0,
	880	'loadxy, test 2 failed')
	881	#self.failUnless(m.userVertices[0].attributes == [10.0,0.0],
	882	# 'loadxy, test 2.2 failed')
	883	self.failUnless(m.userVertices[1].x == 0.0,
	884	'loadxy, test 3 failed')
	885	self.failUnless(m.userVertices[1].y == 1.0,
	886	'loadxy, test 4 failed')
	887	#self.failUnless(m.userVertices[1].attributes == [0.0,10.0],
	888	# 'loadxy, test 5 failed')
	889
[1401]	890	def test_exportPointsFile(self):
[349]	891	a = Vertex (0,0)
	892	b = Vertex (0,3)
	893	c = Vertex (3,3)
	894	d = Vertex (1,2)
	895	e = Vertex (3,1)
	896	f = Vertex (3,1)
	897
[969]	898	s1 = Segment(a,b, tag = 50)
	899	s2 = Segment(b,c, tag = 40)
	900	s3 = Segment(c,a, tag = 30)
[349]	901
	902	r1 = Region(2, 1,tag = 1.3)
	903	h1 = Hole(1,4)
	904	# Warning mesh can't produce this type of data structure its self
	905	m = Mesh(userVertices=[a,b,c,d,e],
	906	userSegments=[s1,s2,s3],
	907	regions=[r1],
	908	holes = [h1])
	909
[1395]	910	fileName = tempfile.mktemp(".xya")
[1401]	911	m.exportPointsFile(fileName)
[349]	912	file = open(fileName)
	913	lFile = file.read().split('\n')
	914	file.close()
[355]	915
[349]	916	os.remove(fileName)
[1401]	917	self.failUnless(lFile[0] == "" and
	918	lFile[1] == "0,0" and
	919	lFile[2] == "0,3" and
	920	lFile[3] == "3,3" and
	921	lFile[4] == "1,2" and
	922	lFile[5] == "3,1"
[349]	923	,
	924	'exported Ascii xya file is wrong')
	925
	926	m.generateMesh("QApz", maxArea = 2.1)
[1401]	927	fileName = tempfile.mktemp(".xya")
	928	m.exportPointsFile(fileName)
[349]	929	file = open(fileName)
	930	lFile = file.read().split('\n')
	931	file.close()
	932	os.remove(fileName)
[1401]	933	self.failUnless(lFile[0] == "" and
	934	lFile[1] == "0.0,0.0" and
	935	lFile[2] == "0.0,3.0" and
	936	lFile[3] == "3.0,3.0" and
	937	lFile[4] == "1.0,2.0" and
	938	lFile[5] == "3.0,1.0" and
	939	lFile[6] == "1.5,1.5"
[349]	940	,
	941	'exported Ascii xya file is wrong')
[1401]	942
	943	def test_exportPointsFilefile2(self):
	944	m = Mesh()
[349]	945
[1401]	946	fileName = tempfile.mktemp(".xya")
	947	m.exportPointsFile(fileName)
	948	file = open(fileName)
	949	lFile = file.read().split('\n')
	950	file.close()
	951
	952	os.remove(fileName)
	953	#print "************* test_mesh exportPointsFilefile"
	954	#print "lFile",lFile
	955	#print "************* test_mesh exportPointsFilefile"
	956	self.failUnless(lFile[0] == ""
	957	,
	958	'exported Ascii xya file is wrong')
	959
[349]	960	def test_strings2ints(self):
	961	list = ["sea","river inlet","","sea","","moat"]
	962	preset = ["moat", "internal boundary"]
	963	[intlist, converter] = segment_strings2ints(list,preset )
	964	self.failUnless(intlist == [2,3 ,0 ,2 ,0 ,0 ]
	965	,
	966	'test_strings2ints produces bad intlist')
	967	self.failUnless(converter == ['moat', 'internal boundary',
	968	'sea', 'river inlet']
	969	,
	970	'test_strings2ints produces bad converter')
	971
	972	def test_ints2strings(self):
	973	list = ["internal boundary","sea","river inlet",
	974	"","sea","","moat","internal boundary"]
	975	outlist = ['internal boundary', 'sea', 'river inlet', 'moat',
	976	'sea', 'moat', 'moat', 'internal boundary']
	977	preset = ["moat", "internal boundary"]
	978	[intlist, converter] = segment_strings2ints(list,preset )
	979	newlist = segment_ints2strings(intlist, converter)
	980	self.failUnless(outlist == newlist
	981	,
	982	'test_strings2ints produces bad intlist')
	983	self.failUnless(converter == ['moat', 'internal boundary',
	984	'sea', 'river inlet']
	985	,
	986	'test_strings2ints produces bad converter')
	987
	988	def test_ints2strings2(self):
	989	list = ["","",""]
	990	preset = ["moat", "internal boundary"]
	991	[intlist, converter] = segment_strings2ints(list,preset )
	992	newlist = segment_ints2strings(intlist, converter)
	993	outlist = ['moat', 'moat', 'moat']
	994	self.failUnless(outlist == newlist
	995	,
	996	'test_strings2ints produces bad intlist')
	997	self.failUnless(converter == ['moat', 'internal boundary']
	998	,
	999	'test_strings2ints produces bad converter')
	1000
	1001
	1002	def test_removeDuplicatedVertices(self):
	1003	a = Vertex (0,0)
	1004	a.index = 0
	1005	b = Vertex (0,3)
	1006	b.index = 1
	1007	c = Vertex (3,3)
	1008	c.index = 2
	1009	d = Vertex (1,1)
	1010	d.index = 3
	1011	e = Vertex (3,1)
	1012	e.index = 4
	1013	f = Vertex (1,1)
	1014	f.index = 5
	1015	g = Vertex (1,1)
	1016	g.index = 6
	1017	inputVerts_noDups = [a,b,c,d,e]
	1018
	1019	m = Mesh(userVertices=[a,b,c,d,e,f,g])
	1020	counter = m.removeDuplicatedUserVertices()
	1021	UserVerts = m.getUserVertices()
	1022
	1023
	1024	self.failUnless(UserVerts == inputVerts_noDups,
	1025	'duplicate verts not removed')
	1026	#for userVert, inputVert in map(None, UserVerts, inputVerts_noDups):
	1027	# self.failUnless(userVert.x == inputVert.x,
	1028	# 'x duplicate verts not removed')
	1029	# self.failUnless(userVert.y == inputVert.y,
	1030	# 'y duplicate verts not removed')
[406]	1031
[1401]	1032
[407]	1033	def test_ungenerateFileLoading(self):
[349]	1034
[406]	1035	fileName = tempfile.mktemp(".txt")
	1036	file = open(fileName,"w")
	1037	file.write(" 1 ?? ??\n\
	1038	0.0 0.0\n\
	1039	1.0 0.0\n\
	1040	1.0 1.0\n\
	1041	0.0 1.0\n\
	1042	0.0 0.0\n\
	1043	END\n\
	1044	2 ?? ??\n\
	1045	10.0 10.0\n\
	1046	10.0 20.0\n\
	1047	20.0 20.0\n\
	1048	10.0 10.0\n\
	1049	END\n\
	1050	END\n")
	1051	file.close()
	1052
	1053
	1054	a = Vertex (0.0, 0.0) #, attributes = [1.1])
	1055	b = Vertex (0.0, 40.0) #, attributes = [1.2])
	1056	c = Vertex (40.0,40.0) #, attributes = [1.3])
	1057	d = Vertex (40.0,0.0) #, attributes = [1.4])
	1058
	1059	s1 = Segment(a,b)
	1060	s2 = Segment(b,c)
	1061	s3 = Segment(c,d)
	1062	s4 = Segment(d,a)
	1063
	1064	m = Mesh(userVertices=[a,b,c,d], userSegments=[s1,s2,s3,s4])
	1065	dict = importUngenerateFile(fileName)
[1185]	1066	os.remove(fileName)
[407]	1067
	1068	tag = "DSG"
	1069	Segment.set_default_tag(tag)
[406]	1070	m.addVertsSegs(dict)
[407]	1071
	1072	# have to reset this , since it's a class attribute
	1073	Segment.set_default_tag("")
[406]	1074
	1075	self.failUnless(len(m.userSegments) ==11,
[407]	1076	'Wrong segment list length.')
[406]	1077	self.failUnless(len(m.userVertices) == 11,
[407]	1078	'Wrong vertex list length.')
[406]	1079	self.failUnless(m.userSegments[10].vertices[0] == m.userVertices[10],
[407]	1080	'bad vertex on segment.')
[406]	1081	self.failUnless(m.userSegments[10].vertices[1] == m.userVertices[8],
[407]	1082	'Bad segment.')
[969]	1083	self.failUnless(m.userSegments[10].tag == tag,
[407]	1084	'wrong tag.')
[431]	1085
	1086	def test_ungenerateFileLoadingII(self):
[406]	1087
[431]	1088	fileName = tempfile.mktemp(".txt")
	1089	file = open(fileName,"w")
	1090	file.write(" 1 ?? ??\n\
	1091	0.0 0.0\n\
	1092	1.0 0.0\n\
	1093	1.0 1.0\n\
	1094	0.0 1.0\n\
	1095	0.0 0.0\n\
	1096	END\n\
	1097	2 ?? ??\n\
	1098	10.0 10.0\n\
	1099	10.0 20.0\n\
	1100	20.0 20.0\n\
	1101	END\n\
	1102	END\n")
	1103	file.close()
	1104
	1105
	1106	a = Vertex (0.0, 0.0) #, attributes = [1.1])
	1107	b = Vertex (0.0, 40.0) #, attributes = [1.2])
	1108	c = Vertex (40.0,40.0) #, attributes = [1.3])
	1109	d = Vertex (40.0,0.0) #, attributes = [1.4])
	1110
	1111	s1 = Segment(a,b)
	1112	s2 = Segment(b,c)
	1113	s3 = Segment(c,d)
	1114	s4 = Segment(d,a)
	1115
	1116	m = Mesh(userVertices=[a,b,c,d], userSegments=[s1,s2,s3,s4])
	1117	dict = importUngenerateFile(fileName)
[1185]	1118	os.remove(fileName)
[431]	1119
	1120	tag = "DSG"
	1121	Segment.set_default_tag(tag)
	1122	m.addVertsSegs(dict)
	1123
	1124	self.failUnless(len(m.userSegments) ==10,
	1125	'Wrong segment list length.')
	1126	self.failUnless(len(m.userVertices) == 11,
	1127	'Wrong vertex list length.')
[1074]	1128
	1129	def test_addVertsSegs(self):
	1130	m = Mesh()
	1131	Segment.set_default_tag("food")
	1132	dict = {}
	1133	dict['points'] = [[0.0, 0.0], [1.0, 0.0], [1.0, 1.0]]
	1134	dict['segments'] = [[0, 1], [1, 2]]
	1135	dict['segment_tags'] = ['','do-op']
	1136	m.addVertsSegs(dict)
	1137	# have to reset this , since it's a class attribute
	1138	Segment.set_default_tag("")
[935]	1139
[1074]	1140
	1141	self.failUnless(len(m.userSegments) ==2,
	1142	'Wrong segment list length.')
	1143	self.failUnless(len(m.userVertices) == 3,
	1144	'Wrong vertex list length.')
	1145	self.failUnless(m.userSegments[0].tag =='food',
	1146	'Wrong segment tag length.')
	1147	self.failUnless(m.userSegments[1].tag =='do-op',
	1148	'Wrong segment tag.')
	1149
[935]	1150	def test_exportASCIImeshfile(self):
	1151
[1001]	1152	#a_att = [5,2]
	1153	#d_att =[4,2]
	1154	#f_att = [3,2]
	1155	#e_att = [2,2]
[935]	1156	a_xy = [0.0, 0.0]
[1001]	1157	a = Vertex ( a_xy[0],a_xy[1]) #, attributes =a_att)
	1158	d = Vertex (0.0, 4.0) #, attributes =d_att)
	1159	f = Vertex (4.0,0.0) #, attributes =f_att)
	1160	e = Vertex (1.0,1.0) #, attributes =e_att)
[935]	1161
[969]	1162	s1 = Segment(a,d, tag = "50")
	1163	s2 = Segment(d,f, tag = "40")
	1164	s3 = Segment(a,f, tag = "30")
	1165	s4 = Segment(a,e, tag = "20")
[935]	1166
	1167	r1 = Region(0.3, 0.3,tag = "1.3", maxArea = 36)
	1168
	1169
	1170	h1 = Hole(0.2,0.6)
[431]	1171
[935]	1172	m = Mesh(userVertices=[a,d,f,e],
	1173	userSegments=[s1,s2,s3,s4],
	1174	regions=[r1],
	1175	holes=[h1])
	1176
	1177	seg = m.getUserSegments()
	1178	points = m.getUserVertices()
	1179	holes = m.getHoles()
	1180	regions = m.getRegions()
[1001]	1181	fileName = tempfile.mktemp(".tsh")
	1182	m.export_mesh_file(fileName)
[935]	1183	#print "***************************fileName", fileName
[1001]	1184	new_m = importMeshFromFile(fileName)
[935]	1185	os.remove(fileName)
[1001]	1186
[935]	1187
[1001]	1188	#print '@@@@@****'
	1189	#print "new_m",new_m
	1190	#print '@@@@@****'
	1191	#print "m",m
	1192	#print '@@@@@****'
	1193
	1194	self.failUnless( new_m == m,
	1195	'loadASCIITestCase failed. test new 1')
	1196
[968]	1197	def test_Mesh2MeshList(self):
	1198
	1199	a_att = [5,2]
	1200	d_att =[4,2]
	1201	f_att = [3,2]
	1202	e_att = [2,2]
	1203	a_xy = [0.0, 0.0]
	1204	a = Vertex ( a_xy[0],a_xy[1]) #, attributes =a_att)
	1205	d = Vertex (0.0, 4.0) #, attributes =d_att)
	1206	f = Vertex (4.0,0.0) #, attributes =f_att)
	1207	e = Vertex (1.0,1.0) #, attributes =e_att)
	1208
[969]	1209	s1 = Segment(a,d, tag = "50")
	1210	s2 = Segment(d,f, tag = "40")
	1211	s3 = Segment(a,f, tag = "30")
	1212	s4 = Segment(a,e, tag = "20")
[968]	1213
	1214	r1 = Region(0.3, 0.3,tag = "1.3", maxArea = 45)
	1215	m = Mesh(userVertices=[a,d,f,e],
	1216	userSegments=[s1,s2,s3,s4],
	1217	regions=[r1])
	1218
	1219	m.generateMesh("QApza2.1")
	1220
	1221	seg = m.getMeshSegments()
	1222	points = m.getMeshVertices()
	1223	dict = m.Mesh2MeshList()
[1001]	1224	#print "dict",dict
[968]	1225	# test not finished...
	1226
	1227	def test_Mesh2IOTriangulationDict(self):
	1228
	1229	a_att = [5,2]
	1230	d_att =[4,2]
	1231	f_att = [3,2]
	1232	e_att = [2,2]
	1233	a_xy = [0.0, 0.0]
	1234	a = Vertex ( a_xy[0],a_xy[1] , attributes =a_att)
	1235	d = Vertex (0.0, 4.0 , attributes =d_att)
	1236	f = Vertex (4.0,0.0 , attributes =f_att)
	1237	e = Vertex (1.0,1.0 , attributes =e_att)
	1238
[969]	1239	s1 = Segment(a,d, tag = "50")
	1240	s2 = Segment(d,f, tag = "40")
	1241	s3 = Segment(a,f, tag = "30")
	1242	s4 = Segment(a,e, tag = "20")
[968]	1243
	1244	r1 = Region(0.3, 0.3,tag = "1.3", maxArea = 45)
	1245	m = Mesh(userVertices=[a,d,f,e],
	1246	userSegments=[s1,s2,s3,s4],
	1247	regions=[r1])
	1248	titles = ['ele','friction']
	1249	m.attributeTitles = titles
	1250	m.generateMesh("QApza2.1")
	1251
	1252	seg = m.getMeshSegments()
	1253	verts = m.getMeshVertices()
	1254	dict = m.Mesh2IOTriangulationDict()
	1255	#print "dict",dict
[935]	1256
[968]	1257	self.failUnless( dict['vertex_attribute_titles'] == titles,
	1258	'test_Mesh2IOTriangulationDict failed. test 1')
	1259	answer = [a_xy,[0.0, 4.0],[4.0,0.0], [1.0,1.0], [2.0,2.0]]
	1260	#print "answer",answer
	1261	#print "dict['vertices']",dict['vertices']
	1262
	1263	self.failUnless( dict['vertices'] == answer,
	1264	'test_Mesh2IOTriangulationDict failed. test 2')
	1265
	1266
	1267	for pimport,pactual,pimpatt in map(None,dict['vertices'],
	1268	verts,dict['vertex_attributes']):
	1269	#assert all_close( pimport, (pactual.x,pactual.y))
	1270	self.failUnless( pimport == [pactual.x,pactual.y],
	1271	'test_Mesh2IOTriangulationDict failed. test 2.1')
	1272	self.failUnless( pimpatt == pactual.attributes,
	1273	'test_Mesh2IOTriangulationDict failed. test 2.2')
	1274	self.failUnless( dict['segments'][0] == [0,1],
	1275	'test_Mesh2IOTriangulationDict failed. test 3')
[969]	1276	for segimp,segactual in map(None,dict['segment_tags'],seg):
	1277	self.failUnless( segimp == segactual.tag,
[968]	1278	'test_Mesh2IOTriangulationDict failed. test 4')
	1279	#print " dict['triangles'][0]", dict['triangles'][0]
	1280	self.failUnless( dict['triangles'][0] == [3,2,4],
	1281	'test_Mesh2IOTriangulationDict failed. test 5')
	1282	self.failUnless( dict['triangle_neighbors'][0] == [-1,2,3],
	1283	'test_Mesh2IOTriangulationDict failed. test 6')
[1183]	1284	self.failUnless( dict['triangle_tags'][0] == "1.3",
[968]	1285	'test_Mesh2IOTriangulationDict failed. test 7')
	1286
	1287
	1288	def test_Mesh2IODict(self):
	1289
	1290	a_att = [5,2]
	1291	d_att =[4,2]
	1292	f_att = [3,2]
	1293	e_att = [2,2]
	1294	a_xy = [0.0, 0.0]
	1295	a = Vertex ( a_xy[0],a_xy[1] , attributes =a_att)
	1296	d = Vertex (0.0, 4.0 , attributes =d_att)
	1297	f = Vertex (4.0,0.0 , attributes =f_att)
	1298	e = Vertex (1.0,1.0 , attributes =e_att)
	1299
[969]	1300	s1 = Segment(a,d, tag = "50")
	1301	s2 = Segment(d,f, tag = "40")
	1302	s3 = Segment(a,f, tag = "30")
	1303	s4 = Segment(a,e, tag = "20")
[968]	1304
	1305	r1 = Region(0.3, 0.3,tag = "1.3", maxArea = 45)
	1306	m = Mesh(userVertices=[a,d,f,e],
	1307	userSegments=[s1,s2,s3,s4],
	1308	regions=[r1])
	1309	titles = ['ele','friction']
	1310	m.attributeTitles = titles
	1311	m.generateMesh("QApza2.1")
	1312
	1313	seg = m.getMeshSegments()
	1314	verts = m.getMeshVertices()
	1315	dict = m.Mesh2IODict()
	1316	#print "dict",dict
	1317
	1318	self.failUnless( dict['vertex_attribute_titles'] == titles,
	1319	'test_Mesh2IOTriangulationDict failed. test 1')
	1320	answer = [a_xy,[0.0, 4.0],[4.0,0.0], [1.0,1.0], [2.0,2.0]]
	1321	#print "answer",answer
	1322	#print "dict['vertices']",dict['vertices']
	1323
	1324	self.failUnless( dict['vertices'] == answer,
	1325	'test_Mesh2IOTriangulationDict failed. test 2')
	1326
	1327
	1328	for pimport,pactual,pimpatt in map(None,dict['vertices'],
	1329	verts,dict['vertex_attributes']):
	1330	#assert all_close( pimport, (pactual.x,pactual.y))
	1331	self.failUnless( pimport == [pactual.x,pactual.y],
	1332	'test_Mesh2IODict failed. test 2.1')
	1333	self.failUnless( pimpatt == pactual.attributes,
	1334	'test_Mesh2IODict failed. test 2.2')
	1335	self.failUnless( dict['segments'][0] == [0,1],
	1336	'test_Mesh2IODict failed. test 3')
[969]	1337	for segimp,segactual in map(None,dict['segment_tags'],seg):
	1338	self.failUnless( segimp == segactual.tag,
[968]	1339	'test_Mesh2IODict failed. test 4')
	1340	#print " dict['triangles'][0]", dict['triangles'][0]
	1341	self.failUnless( dict['triangles'][0] == [3,2,4],
	1342	'test_Mesh2IODict failed. test 5')
	1343	self.failUnless( dict['triangle_neighbors'][0] == [-1,2,3],
	1344	'test_Mesh2IODict failed. test 6')
[1183]	1345	self.failUnless( dict['triangle_tags'][0] == "1.3",
[968]	1346	'test_Mesh2IODict failed. test 7')
	1347
	1348	seg = m.getUserSegments()
	1349	points = m.getUserVertices()
	1350	holes = m.getHoles()
	1351	regions = m.getRegions()
	1352
	1353	for pimport,pactual,pimpatt in map(None,dict['points'],points,dict['point_attributes']):
	1354	self.failUnless( pimport == [pactual.x,pactual.y],
	1355	'test_Mesh2IODict failed. test 1')
	1356	self.failUnless( pimpatt == pactual.attributes,
	1357	'test_Mesh2IODict failed. test 1.1')
	1358	self.failUnless( dict['outline_segments'][0] == [0,1],
	1359	'test_Mesh2IODict failed. test 3')
[969]	1360	for segimp,segactual in map(None,dict['outline_segment_tags'],seg):
	1361	self.failUnless( segimp == segactual.tag,
[968]	1362	'test_Mesh2IODict failed. test 4')
	1363	for holeimp,holeactual in map(None,dict['holes'],holes):
	1364	self.failUnless( holeimp == [holeactual.x,holeactual.y],
	1365	'test_Mesh2IODict failed. test 5')
	1366
[969]	1367	for regimp,regactual,regattimp, regmaxarea in map(None,dict['regions'],regions, dict['region_tags'], dict['region_max_areas']):
[968]	1368	self.failUnless( regimp == [regactual.x,regactual.y],
	1369	'loadASCIITestCase failed. test 6')
	1370	self.failUnless( regattimp == regactual.getTag(),
	1371	'loadASCIITestCase failed. test 7')
	1372	self.failUnless( regmaxarea == regactual.getMaxArea(),
	1373	'loadASCIITestCase failed. test 7')
	1374
	1375
	1376
	1377	def test_Mesh2IOOutlineDict(self):
	1378
	1379	a_att = [5,2]
	1380	d_att =[4,2]
	1381	f_att = [3,2]
	1382	e_att = [2,2]
	1383	a_xy = [0.0, 0.0]
	1384	a = Vertex ( a_xy[0],a_xy[1] , attributes =a_att)
	1385	d = Vertex (0.0, 4.0 , attributes =d_att)
	1386	f = Vertex (4.0,0.0 , attributes =f_att)
	1387	e = Vertex (1.0,1.0 , attributes =e_att)
	1388
[969]	1389	s1 = Segment(a,d, tag = "50")
	1390	s2 = Segment(d,f, tag = "40")
	1391	s3 = Segment(a,f, tag = "30")
	1392	s4 = Segment(a,e, tag = "20")
[968]	1393
	1394	r1 = Region(0.3, 0.3,tag = "1.3", maxArea = 45)
	1395	m = Mesh(userVertices=[a,d,f,e],
	1396	userSegments=[s1,s2,s3,s4],
	1397	regions=[r1])
	1398	titles = ['ele','friction']
	1399	m.attributeTitles = titles
	1400	m.generateMesh("QApza2.1")
	1401
	1402	seg = m.getMeshSegments()
	1403	verts = m.getMeshVertices()
	1404	dict = m.Mesh2IOOutlineDict()
	1405
	1406	seg = m.getUserSegments()
	1407	points = m.getUserVertices()
	1408	holes = m.getHoles()
	1409	regions = m.getRegions()
	1410
	1411	for pimport,pactual,pimpatt in map(None,dict['points'],points,dict['point_attributes']):
	1412	self.failUnless( pimport == [pactual.x,pactual.y],
	1413	'loadASCIITestCase failed. test 1')
	1414	self.failUnless( pimpatt == pactual.attributes,
	1415	'loadASCIITestCase failed. test 1.1')
	1416	self.failUnless( dict['outline_segments'][0] == [0,1],
	1417	'loadASCIITestCase failed. test 3')
[969]	1418	for segimp,segactual in map(None,dict['outline_segment_tags'],seg):
	1419	self.failUnless( segimp == segactual.tag,
[968]	1420	'loadASCIITestCase failed. test 4')
	1421	for holeimp,holeactual in map(None,dict['holes'],holes):
	1422	self.failUnless( holeimp == [holeactual.x,holeactual.y],
	1423	'loadASCIITestCase failed. test 5')
	1424	#for regimp,regactual in map(None,dict['regions'],regions):
	1425	# self.failUnless( [regimp[0],regimp[1]]==[regactual.x,regactual.y],
	1426	# 'loadASCIITestCase failed. test 6')
	1427	# self.failUnless( regimp[2] == regactual.getTag(),
	1428	# 'loadASCIITestCase failed. test 7')
	1429	#self.failUnless( regimp[3] == regactual.getMaxArea(),
	1430	# 'loadASCIITestCase failed. test 7')
	1431
	1432
[969]	1433	for regimp,regactual,regattimp, regmaxarea in map(None,dict['regions'],regions, dict['region_tags'], dict['region_max_areas']):
[968]	1434	self.failUnless( regimp == [regactual.x,regactual.y],
	1435	'loadASCIITestCase failed. test 6')
	1436	self.failUnless( regattimp == regactual.getTag(),
	1437	'loadASCIITestCase failed. test 7')
	1438	self.failUnless( regmaxarea == regactual.getMaxArea(),
	1439	'loadASCIITestCase failed. test 7')
[1361]	1440
	1441
	1442	#___________begining of Peters tests
	1443
	1444	def test_set_stuff(self):
	1445	"""
	1446	Documentation
	1447	"""
	1448	#making a test mesh
	1449	p0=[0.,2.]
	1450	p1=[1.,2.]
	1451	p2=[0.,1.]
	1452	p3=[1.,1.]
	1453	p4=[0.,0.]
	1454	p5=[2.,0.]
	1455	p6=[-1.,1.]
	1456	point_list = [p0,p1,p2,p3,p4,p5,p6]
	1457
	1458	a0=[0]
	1459	a1=[0]
	1460	a2=[100]
	1461	a3=[0]
	1462	a4=[0]
	1463	a5=[0]
	1464	a6=[0]
	1465	attribute=[a0,a1,a2,a3,a4,a5,a6]
	1466
	1467	t0=[0,3,1]
	1468	t1=[0,2,3]
	1469	t2=[2,4,3]
	1470	t3=[4,5,3]
	1471	t4=[1,3,5]
	1472	t5=[2,6,4]
	1473
	1474	n0=[4,-1,2]
	1475	n1=[2,0,-1]
	1476	n2=[3,1,5]
	1477	n3=[4,2,-1]
	1478	n4=[3,-1,0]
	1479	n5=[-1,2,-1]
	1480
	1481	tri_list = [t0,t1,t2,t3,t4,t5]
	1482	n_list = [n0,n1,n2,n3,n4,n5]
	1483	for i in range(6):
	1484	for j in (0,1,2):
	1485	a=attribute[tri_list[i][j]]
	1486	tri_list[i][j]=point_list[tri_list[i][j]]
	1487	tri_list[i][j]=Vertex(tri_list[i][j][0]\
	1488	,tri_list[i][j][1],a)
	1489	neighbours=n_list[i]
	1490	tri_list[i]=Triangle(tri_list[i][0],\
	1491	tri_list[i][1],tri_list[i][2]\
	1492	,neighbors=neighbours)
	1493
	1494	#testing selectAll
	1495	mesh = Mesh()
	1496	mesh.attributeTitles=['attrib']
	1497
	1498	mesh.meshTriangles=tri_list
	1499
	1500	mesh.selectAllTriangles()
	1501	A=mesh.sets[mesh.setID['All']]
	1502	assert list_comp(tri_list,A)
	1503
	1504	#testing threshold
	1505	mesh = Mesh()
	1506	mesh.attributeTitles=['attrib']
	1507
	1508	mesh.meshTriangles=tri_list
	1509	mesh.selectAllTriangles()
	1510	mesh.threshold('All',min=30,max=35,attribute_name = 'attrib')
	1511	A = [tri_list[1],tri_list[2],tri_list[5]]
	1512	B = mesh.sets[mesh.setID['All']]
	1513	assert list_comp(A,B)
	1514
	1515
	1516	A = [tri_list[3],tri_list[2],tri_list[5]]
	1517	assert not list_comp(A,B)
	1518
	1519	#testing
	1520
	1521	def test_Discretised_Tuple_Set_rounding(self):
	1522	#This is the hardest bit of DST
	1523	from mesh import Discretised_Tuple_Set
	1524
	1525	tol = 0.1
	1526	a=Discretised_Tuple_Set(p_rel=1,t_rel= tol)
	1527	m = 0.541
	1528	m_up = 0.6
	1529	m_down = 0.5
	1530	assert m_up == a.round_up_rel(m)
	1531	assert m_down == a.round_down_rel(m)
	1532
	1533	tol = 0.1
	1534	a=Discretised_Tuple_Set(p_rel=1,t_rel = tol)
	1535	m = 0.539
	1536	m_up = 0.5
	1537	m_down = 0.5
	1538	assert m_up == a.round_up_rel(m)
	1539	assert m_down == a.round_down_rel(m)
	1540
	1541	tol = 0.5
	1542	a=Discretised_Tuple_Set(p_rel=1,t_rel = tol)
	1543
	1544
	1545	m = 0.6
	1546	m_up = 0.7
	1547	m_down = 0.5
	1548	assert m_up == a.round_up_rel(m)
	1549	assert m_down == a.round_down_rel(m)
	1550
	1551	m = 0.599
	1552	m_up = 0.6
	1553	m_down = 0.5
	1554	assert m_up == a.round_up_rel(m)
	1555	assert m_down == a.round_down_rel(m)
	1556
	1557	def test_Discretised_Tuple_Set_get(self):
	1558	from mesh import Discretised_Tuple_Set
	1559	tol = 0.25
	1560	a=Discretised_Tuple_Set(p_rel=1,t_rel = tol)
	1561	b = (1.1,1.1)
	1562	a.append(b)
	1563	list = [(1.2,1.),(1.,1.),(1.,1.2),(1.2,1.2)]
	1564	for key in list:
	1565	assert a[key][0]==b
	1566	assert len(a[key])==1
	1567
	1568	c = (2.1,1.)
	1569	a.append(c)
	1570	assert a[(2.,1.)][0]==c
	1571	assert a[(2.2,1.)][0]==c
	1572
	1573	def test_mapped_Discretised_Tuple_Set(self):
	1574
	1575	from mesh import Mapped_Discretised_Tuple_Set
	1576	from mesh import Discretised_Tuple_Set
	1577
	1578	def map(sequence):
	1579	return [len(sequence)]
	1580
	1581	tol = 0.5
	1582	a=Mapped_Discretised_Tuple_Set(map,p_rel=1,t_rel = tol)
	1583	b = range(20)
	1584	a.append(b)
	1585	assert b in a[range(17)]
	1586	assert b in a[range(22)]
	1587
	1588	tol = 0.01
	1589	a=Mapped_Discretised_Tuple_Set(map,p_rel=1,t_rel = tol)
	1590	b = range(20)
	1591	a.append(b)
	1592	assert b in a[range(20)]
	1593	assert b in a[range(19)]
	1594	assert not range(17) in a
	1595
	1596
	1597	def list_comp(A,B):
	1598	yes = len(A)==len(B)
	1599	for item in A:
	1600	if not item in B:
	1601	yes = False
	1602	return yes
	1603
	1604	#___________end of Peters tests
[968]	1605
[349]	1606	#-------------------------------------------------------------
	1607	if __name__ == "__main__":
[1034]	1608	suite = unittest.makeSuite(meshTestCase,'test')
[1074]	1609	#suite = unittest.makeSuite(meshTestCase,'test_asciiFile')
[702]	1610	runner = unittest.TextTestRunner() #verbosity=2)
[349]	1611	runner.run(suite)
	1612

Note: See TracBrowser for help on using the repository browser.

Download in other formats: