[349] | 1 | #!/usr/bin/env python |
---|
| 2 | # |
---|
| 3 | |
---|
| 4 | import unittest |
---|
| 5 | import triang |
---|
| 6 | |
---|
| 7 | class triangTestCase(unittest.TestCase): |
---|
| 8 | def setUp(self): |
---|
| 9 | pass |
---|
| 10 | |
---|
| 11 | def tearDown(self): |
---|
| 12 | pass |
---|
[604] | 13 | |
---|
[349] | 14 | def testrectangle(self): |
---|
| 15 | |
---|
| 16 | points = [] |
---|
| 17 | seglist = [] |
---|
[604] | 18 | segattlist = [] |
---|
[349] | 19 | holelist = [] |
---|
| 20 | regionlist = [] |
---|
| 21 | |
---|
| 22 | points = [(0.0,0.0),(0.0,10.0),(3.0,0.0),(3.0,10.0)] |
---|
[651] | 23 | # This will cause a seg fault |
---|
| 24 | #points = [[0.0,0.0],[0.0,10.0],[3.0,0.0],[3.0,10.0]] |
---|
[604] | 25 | |
---|
| 26 | # automatically generate pointattlist |
---|
| 27 | pointattlist = [] |
---|
| 28 | for point in points: |
---|
| 29 | pointattlist.append([]) |
---|
| 30 | #print "pointattlist",pointattlist |
---|
| 31 | #pointattlist = [[],[],[],[]] |
---|
[349] | 32 | trilist = [] |
---|
[604] | 33 | mode = "Qzcn" |
---|
[349] | 34 | data = triang.genMesh(points,seglist,holelist,regionlist,pointattlist,segattlist,trilist, mode) |
---|
[604] | 35 | self.failUnless(data['generatedtrianglelist'] ==[(1, 0, 2), (2, 3, 1)], |
---|
| 36 | 'generatedtrianglelist is wrong!') |
---|
| 37 | #print "data['generatedsegmentlist']",data['generatedsegmentlist'] |
---|
| 38 | self.failUnless(data['generatedsegmentlist'] ==[(2, 0), (3, 2), (1, 3), (0, 1)] |
---|
| 39 | , 'generatedsegmentlist is wrong!') |
---|
| 40 | self.failUnless(data['generatedpointlist'] ==[(0.0, 0.0), (0.0, 10.0), (3.0, 0.0), (3.0, 10.0)], |
---|
| 41 | ' is wrong!') |
---|
[349] | 42 | |
---|
[604] | 43 | #print "data['generatedtriangleneighborlist']",data['generatedtriangleneighborlist'] |
---|
| 44 | self.failUnless(data['generatedtriangleneighborlist'] ==[(-1, 1, -1), (-1, 0, -1)], |
---|
| 45 | 'generatedtriangleneighborlist is wrong!') |
---|
[349] | 46 | |
---|
| 47 | |
---|
[604] | 48 | def testrectangleII(self): |
---|
[349] | 49 | |
---|
| 50 | points = [] |
---|
| 51 | seglist = [] |
---|
| 52 | holelist = [] |
---|
| 53 | regionlist = [] |
---|
| 54 | |
---|
| 55 | points = [(0.0,0.0),(0.0,10.0),(3.0,0.0),(3.0,10.0)] |
---|
| 56 | pointattlist = [[],[],[],[]] |
---|
| 57 | regionlist.append( (1.2,1.2,5.0) ) |
---|
| 58 | seglist = [(0,1),(1,3),(3,2),(2,0)] |
---|
| 59 | segattlist = [0,0,0,0] |
---|
| 60 | trilist = [] |
---|
| 61 | mode = "Qzp" |
---|
| 62 | data = triang.genMesh(points,seglist,holelist,regionlist,pointattlist,segattlist,trilist, mode) |
---|
| 63 | |
---|
| 64 | self.failUnless(data['generatedtrianglelist'] ==[(1, 0, 2), (2, 3, 1)], |
---|
| 65 | 'trianglelist is wrong!') |
---|
| 66 | self.failUnless(data['generatedsegmentlist'] ==[(0, 1), (1, 3), (3, 2), (2, 0)], |
---|
| 67 | 'segmentlist is wrong!') |
---|
| 68 | self.failUnless(data['generatedpointlist'] ==[(0.0, 0.0), (0.0, 10.0), (3.0, 0.0), (3.0, 10.0)], |
---|
| 69 | ' is wrong!') |
---|
| 70 | |
---|
| 71 | def testbad_region(self): |
---|
| 72 | |
---|
| 73 | points = [] |
---|
| 74 | seglist = [] |
---|
| 75 | holelist = [] |
---|
| 76 | regionlist = [(1.2,1.2)] |
---|
| 77 | |
---|
| 78 | points = [(0.0,0.0),(0.0,10.0),(3.0,0.0),(3.0,10.0)] |
---|
| 79 | pointattlist = [[],[],[],[]] |
---|
| 80 | seglist = [(0,1),(1,3),(3,2),(2,0)] |
---|
| 81 | segattlist = [0,0,0,0] |
---|
| 82 | trilist = [] |
---|
| 83 | mode = "Qzpn" |
---|
| 84 | try: |
---|
| 85 | data = triang.genMesh(points,seglist,holelist,regionlist,pointattlist,segattlist, trilist, mode) |
---|
| 86 | |
---|
| 87 | except TypeError: |
---|
| 88 | pass |
---|
| 89 | else: |
---|
| 90 | self.failUnless(0 ==1, |
---|
| 91 | 'bad region list did not raise error!') |
---|
| 92 | |
---|
| 93 | def testregion_with_maxarea(self): |
---|
| 94 | |
---|
| 95 | points = [] |
---|
| 96 | seglist = [] |
---|
| 97 | holelist = [] |
---|
| 98 | regionlist = [(3,1,1.0)] |
---|
| 99 | |
---|
| 100 | points = [(0.0,0.0),(6.0,0.0),(6.0,6.0),(0.0,6.0)] |
---|
| 101 | pointattlist = [[],[],[],[]] |
---|
| 102 | seglist = [(0,1),(1,2),(3,2),(3,0),(0,2)] |
---|
| 103 | segattlist = [0,0,0,0,0] |
---|
| 104 | trilist = [] |
---|
| 105 | mode = "Qzpna36a" |
---|
| 106 | data = triang.genMesh(points,seglist,holelist,regionlist,pointattlist,segattlist, trilist, mode) |
---|
| 107 | self.failUnless(len(data['generatedtrianglelist']) == 2, |
---|
| 108 | 'testregion_with_maxarea 1: # of tris is wrong!') |
---|
| 109 | ## Another test case |
---|
| 110 | regionlist = [(3,1,1.0),(1,3,1.0,8.0)] |
---|
| 111 | mode = "Qzpna36a" |
---|
| 112 | data = triang.genMesh(points,seglist,holelist,regionlist,pointattlist,segattlist, trilist, mode) |
---|
| 113 | self.failUnless(len(data['generatedtrianglelist']) == 6, |
---|
| 114 | 'testregion_with_maxarea 2: # of tris is wrong!') |
---|
| 115 | ## Another test case |
---|
| 116 | regionlist = [(3,1,1.0,8.0),(1,3,1.0,8.0)] |
---|
| 117 | mode = "Qzpna36a" |
---|
| 118 | data = triang.genMesh(points,seglist,holelist,regionlist,pointattlist,segattlist, trilist, mode) |
---|
| 119 | self.failUnless(len(data['generatedtrianglelist']) == 8, |
---|
| 120 | 'testregion_with_maxarea 3: # of tris is wrong!') |
---|
| 121 | |
---|
| 122 | ## Another test case |
---|
| 123 | regionlist = [(3,1,1.0),(1,3,1.0,8.0)] |
---|
| 124 | mode = "Qzpna8a" |
---|
| 125 | data = triang.genMesh(points,seglist,holelist,regionlist,pointattlist,segattlist, trilist, mode) |
---|
| 126 | self.failUnless(len(data['generatedtrianglelist']) == 8, |
---|
| 127 | 'testregion_with_maxarea 4: # of tris is wrong!') |
---|
| 128 | |
---|
| 129 | |
---|
| 130 | def testbad_point(self): |
---|
| 131 | |
---|
| 132 | holelist = [] |
---|
| 133 | regionlist = [] |
---|
| 134 | |
---|
| 135 | points = [(0.0,0.0),(0.0,10.0),(3.0,0.0),(3.0)] |
---|
| 136 | pointattlist = [[],[],[],[]] |
---|
| 137 | seglist = [(0,1),(1,3),(3,2),(2,0)] |
---|
| 138 | segattlist = [0,0,0,0] |
---|
| 139 | trilist = [] |
---|
| 140 | mode = "Qzpn" |
---|
| 141 | try: |
---|
| 142 | data = triang.genMesh(points,seglist,holelist,regionlist,pointattlist,segattlist, trilist, mode) |
---|
| 143 | |
---|
| 144 | except TypeError: |
---|
| 145 | pass |
---|
| 146 | else: |
---|
| 147 | self.failUnless(0 ==1, |
---|
| 148 | 'bad point list did not raise error!') |
---|
| 149 | |
---|
| 150 | def testbad_hole(self): |
---|
| 151 | |
---|
| 152 | holelist = [(9.0)] |
---|
| 153 | regionlist = [] |
---|
| 154 | |
---|
| 155 | points = [(0.0,0.0),(0.0,10.0),(3.0,0.0),(3.0,10.0)] |
---|
| 156 | pointattlist = [[],[],[],[]] |
---|
| 157 | seglist = [(0,1),(1,3),(3,2),(2,0)] |
---|
| 158 | segattlist = [0,0,0,0] |
---|
| 159 | trilist = [] |
---|
| 160 | mode = "Qzpn" |
---|
| 161 | try: |
---|
| 162 | data = triang.genMesh(points,seglist,holelist,regionlist,pointattlist,segattlist, trilist, mode) |
---|
| 163 | |
---|
| 164 | except TypeError: |
---|
| 165 | pass |
---|
| 166 | else: |
---|
| 167 | self.failUnless(0 ==1, |
---|
| 168 | 'bad hole list did not raise error!') |
---|
| 169 | |
---|
| 170 | def testbad_segment(self): |
---|
| 171 | |
---|
| 172 | holelist = [] |
---|
| 173 | regionlist = [] |
---|
| 174 | |
---|
| 175 | points = [(0.0,0.0),(0.0,10.0),(3.0,0.0),(3.0,10.0)] |
---|
| 176 | pointattlist = [[],[],[],[]] |
---|
| 177 | seglist = [(0,1),(1,3),(3,2),(2)] |
---|
| 178 | segattlist = [0,0,0,0] |
---|
| 179 | trilist = [] |
---|
| 180 | mode = "Qzpn" |
---|
| 181 | try: |
---|
| 182 | data = triang.genMesh(points,seglist,holelist,regionlist,pointattlist,segattlist, trilist, mode) |
---|
| 183 | |
---|
| 184 | except TypeError: |
---|
| 185 | pass |
---|
| 186 | else: |
---|
| 187 | self.failUnless(0 ==1, |
---|
| 188 | 'bad segment list did not raise error!') |
---|
| 189 | |
---|
| 190 | def testbad_segattlist(self): |
---|
| 191 | |
---|
| 192 | holelist = [] |
---|
| 193 | regionlist = [] |
---|
| 194 | |
---|
| 195 | points = [(0.0,0.0),(0.0,10.0),(3.0,0.0),(3.0,10.0)] |
---|
| 196 | pointattlist = [[],[],[],[]] |
---|
| 197 | seglist = [(0,1),(1,3),(3,2),(2,0)] |
---|
| 198 | segattlist = [0,0] |
---|
| 199 | trilist = [] |
---|
| 200 | mode = "Qzpn" |
---|
| 201 | try: |
---|
| 202 | data = triang.genMesh(points,seglist,holelist,regionlist,pointattlist,segattlist, trilist, mode) |
---|
| 203 | |
---|
| 204 | self.failUnless(data['trianglelist'] ==[(1, 0, 2), (2, 3, 1)], |
---|
| 205 | 'trianglelist is wrong!') |
---|
| 206 | except TypeError: |
---|
| 207 | pass |
---|
| 208 | else: |
---|
| 209 | self.failUnless(0 ==1, |
---|
| 210 | 'bad segment attribute list did not raise error!') |
---|
| 211 | |
---|
| 212 | def testrectangle_regions(self): |
---|
| 213 | |
---|
| 214 | points = [] |
---|
| 215 | seglist = [] |
---|
| 216 | holelist = [] |
---|
| 217 | regionlist = [] |
---|
| 218 | |
---|
| 219 | points = [(0.0,0.0),(0.0,10.0),(3.0,0.0),(3.0,10.0)] |
---|
| 220 | pointattlist = [[],[],[],[]] |
---|
| 221 | #it seems that |
---|
| 222 | #triangle only associates one region with a triangle |
---|
| 223 | regionlist.append( (1.3,1.3,88.33) ) |
---|
| 224 | regionlist.append( (1.2,1.2,77,55) ) |
---|
| 225 | seglist = [(0,1),(1,3),(3,2),(2,0)] |
---|
| 226 | segattlist = [0,0,0,0] |
---|
| 227 | trilist = [] |
---|
| 228 | mode = "QAzpq" |
---|
| 229 | data = triang.genMesh(points,seglist,holelist,regionlist,pointattlist,segattlist, trilist, mode) |
---|
| 230 | |
---|
| 231 | self.failUnless(data['generatedtriangleattributelist'] ==[[77.0], [77.0], [77.0], [77.0]], |
---|
| 232 | 'triangleattributelist is wrong!') |
---|
| 233 | |
---|
| 234 | |
---|
| 235 | if __name__ == "__main__": |
---|
| 236 | |
---|
| 237 | suite = unittest.makeSuite(triangTestCase,'test') |
---|
| 238 | runner = unittest.TextTestRunner() |
---|
| 239 | runner.run(suite) |
---|