Changeset 5951
- Timestamp:
- Nov 12, 2008, 3:16:51 PM (16 years ago)
- Location:
- anuga_core/source_numpy_conversion/anuga
- Files:
-
- 6 edited
Legend:
- Unmodified
- Added
- Removed
-
anuga_core/source_numpy_conversion/anuga/load_mesh/test_loadASCII.py
r5910 r5951 179 179 #print "*(*(" 180 180 181 self.failUnless(numpy.array(meshDict['vertices']) == 182 numpy.array(loadedDict['vertices']), 183 'test_export_mesh_file failed. Test 1') 184 self.failUnless(numpy.array(meshDict['triangles']) == 185 numpy.array(loadedDict['triangles']), 186 'test_export_mesh_file failed. Test 2') 187 self.failUnless(numpy.array(meshDict['segments']) == 188 numpy.array(loadedDict['segments']), 189 'test_export_mesh_file failed. Test 3') 190 self.failUnless(numpy.array(meshDict['triangle_tags']) == 191 numpy.array(loadedDict['triangle_tags']), 181 self.failUnless(numpy.alltrue(numpy.array(meshDict['vertices']) == 182 numpy.array(loadedDict['vertices']) 183 ), 184 'test_export_mesh_file failed. Test 1') 185 self.failUnless(numpy.alltrue(numpy.array(meshDict['triangles']) == 186 numpy.array(loadedDict['triangles'])), 187 'test_export_mesh_file failed. Test 2') 188 self.failUnless(numpy.alltrue(numpy.array(meshDict['segments']) == 189 numpy.array(loadedDict['segments']) 190 ), 191 'test_export_mesh_file failed. Test 3') 192 self.failUnless(numpy.alltrue(numpy.array(meshDict['triangle_tags']) == 193 numpy.array(loadedDict['triangle_tags']) 194 ), 192 195 'test_export_mesh_file failed. Test 4') 193 196 194 197 self.failUnless(meshDict['vertex_attributes'] == 195 198 loadedDict['vertex_attributes'], 196 'test_export_mesh_file failed. Test 5') 197 self.failUnless(numpy.array(meshDict['triangle_neighbors']) == 198 numpy.array(loadedDict['triangle_neighbors']), 199 'test_export_mesh_file failed. Test 6') 200 self.failUnless(numpy.array(meshDict['segment_tags']) == 201 numpy.array(loadedDict['segment_tags']), 202 'test_export_mesh_file failed. Test 7') 203 self.failUnless(numpy.array(meshDict['vertex_attribute_titles']) == 204 numpy.array(loadedDict['vertex_attribute_titles']), 205 'test_export_mesh_file failed. Test 8') 206 self.failUnless(numpy.array(meshDict['geo_reference']) == 207 numpy.array(loadedDict['geo_reference']), 208 'test_export_mesh_file failed. Test 9') 199 'test_export_mesh_file failed. Test 5') 200 self.failUnless(numpy.alltrue(numpy.array(meshDict['triangle_neighbors']) == 201 numpy.array(loadedDict['triangle_neighbors']) 202 ), 203 'test_export_mesh_file failed. Test 6') 204 self.failUnless(numpy.alltrue(numpy.array(meshDict['segment_tags']) == 205 numpy.array(loadedDict['segment_tags']) 206 ), 207 'test_export_mesh_file failed. Test 7') 208 self.failUnless(numpy.alltrue(numpy.array(meshDict['vertex_attribute_titles']) == 209 numpy.array(loadedDict['vertex_attribute_titles']) 210 ), 211 'test_export_mesh_file failed. Test 8') 212 self.failUnless(numpy.alltrue(numpy.array(meshDict['geo_reference']) == 213 numpy.array(loadedDict['geo_reference']) 214 ), 215 'test_export_mesh_file failed. Test 9') 209 216 210 217 os.remove(fileName) -
anuga_core/source_numpy_conversion/anuga/mesh_engine/test_generate_mesh.py
r5911 r5951 71 71 pointattlist,segattlist, mode, points) 72 72 73 self.failUnless(data['generatedtrianglelist'] ==[(1, 0, 2), (2, 3, 1)], 74 'trianglelist is wrong!') 75 self.failUnless(data['generatedsegmentlist'] ==[(0, 1), (1, 3), 76 (3, 2), (2, 0)], 77 'segmentlist is wrong!') 78 self.failUnless(data['generatedpointlist'] ==[(0.0, 0.0), (0.0, 10.0), 79 (3.0, 0.0), (3.0, 10.0)], 73 self.failUnless(numpy.alltrue(data['generatedtrianglelist'] == 74 [(1, 0, 2), (2, 3, 1)] 75 ), 76 'trianglelist is wrong!') 77 self.failUnless(numpy.alltrue(data['generatedsegmentlist'] == 78 [(0, 1), (1, 3), (3, 2), (2, 0)] 79 ), 80 'segmentlist is wrong!') 81 self.failUnless(numpy.alltrue(data['generatedpointlist'] == 82 [(0.0, 0.0), (0.0, 10.0), (3.0, 0.0), (3.0, 10.0)] 83 ), 80 84 ' is wrong!') 81 85 … … 98 102 pointattlist,segattlist, mode, points) 99 103 100 self.failUnless(data['generatedtrianglelist'] ==[(1, 0, 2), (2, 3, 1)], 101 'trianglelist is wrong!') 102 self.failUnless(data['generatedsegmentlist'] ==[(0, 1), (1, 3), 103 (3, 2), (2, 0)], 104 'segmentlist is wrong!') 105 self.failUnless(data['generatedpointlist'] ==[(0.0, 0.0), (0.0, 10.0), 106 (3.0, 0.0), (3.0, 10.0)], 104 self.failUnless(numpy.alltrue(data['generatedtrianglelist'] == 105 [(1, 0, 2), (2, 3, 1)]), 106 'trianglelist is wrong!') 107 self.failUnless(numpy.alltrue(data['generatedsegmentlist'] == 108 [(0, 1), (1, 3), (3, 2), (2, 0)]), 109 'segmentlist is wrong!') 110 self.failUnless(numpy.alltrue(data['generatedpointlist'] == 111 [(0.0, 0.0), (0.0, 10.0), (3.0, 0.0), (3.0, 10.0)]), 107 112 ' is wrong!') 108 113 … … 172 177 pointattlist,segattlist, mode, points) 173 178 174 self.failUnless(data['generatedtrianglelist'] ==[(1, 0, 2), (2, 3, 1)], 175 'trianglelist is wrong!') 176 self.failUnless(data['generatedsegmentlist'] ==[(0, 1), (1, 3), 177 (3, 2), (2, 0)], 178 'segmentlist is wrong!') 179 self.failUnless(data['generatedpointlist'] ==[(0.0, 0.0), (0.0, 10.0), 180 (3.0, 0.0), (3.0, 10.0)], 179 self.failUnless(numpy.alltrue(data['generatedtrianglelist'] == 180 [(1, 0, 2), (2, 3, 1)]), 181 'trianglelist is wrong!') 182 self.failUnless(numpy.alltrue(data['generatedsegmentlist'] == 183 [(0, 1), (1, 3), (3, 2), (2, 0)]), 184 'segmentlist is wrong!') 185 self.failUnless(numpy.alltrue(data['generatedpointlist'] == 186 [(0.0, 0.0), (0.0, 10.0), (3.0, 0.0), (3.0, 10.0)]), 181 187 'generatedpointlist is wrong!') 182 self.failUnless( data['generatedsegmentmarkerlist'] ==[1,2,3,4],188 self.failUnless(numpy.alltrue(data['generatedsegmentmarkerlist'] == [1,2,3,4]), 183 189 'generatedsegmentmarkerlist is wrong!') 184 190 … … 351 357 pointattlist,segattlist, mode, points) 352 358 353 self.failUnless(data['generatedtriangleattributelist'] ==[[77.0], [77.0], [77.0], [77.0]], 359 self.failUnless(numpy.alltrue(data['generatedtriangleattributelist'] == 360 [[77.0], [77.0], [77.0], [77.0]]), 354 361 'triangleattributelist is wrong!') 355 362 … … 373 380 pointattlist,segattlist, mode, points) 374 381 375 self.failUnless(data['generatedtriangleattributelist'] ==[[77.0], [77.0], [77.0], [77.0]], 382 self.failUnless(numpy.alltrue(data['generatedtriangleattributelist'] == 383 [[77.0], [77.0], [77.0], [77.0]]), 376 384 'triangleattributelist is wrong!') 377 385 … … 599 607 #print "data", data 600 608 601 self.failUnless(data['generatedtrianglelist'] ==[(1, 0, 2), (2, 3, 1)], 602 'trianglelist is wrong!') 603 self.failUnless(data['generatedsegmentlist'] ==[(0, 1), (1, 3), 604 (3, 2), (2, 0)], 605 'segmentlist is wrong!') 606 self.failUnless(data['generatedpointlist'] ==[(0.0, 0.0), (0.0, 10.0), 607 (3.0, 0.0), (3.0, 10.0)], 608 ' is wrong!') 609 self.failUnless(data['generatedtriangleattributelist'] == [[tri_tag], 610 [tri_tag]], 611 ' is wrong!') 612 613 self.failUnless(data['generatedsegmentlist'] == seglist, 614 ' is wrong!') 615 self.failUnless(data['generatedsegmentmarkerlist'] == segattlist, 609 self.failUnless(numpy.alltrue(data['generatedtrianglelist'] == 610 [(1, 0, 2), (2, 3, 1)] 611 ), 612 'trianglelist is wrong!') 613 self.failUnless(numpy.alltrue(data['generatedsegmentlist'] == 614 [(0, 1), (1, 3), (3, 2), (2, 0)] 615 ), 616 'segmentlist is wrong!') 617 self.failUnless(numpy.alltrue(data['generatedpointlist'] == 618 [(0.0, 0.0), (0.0, 10.0), (3.0, 0.0), (3.0, 10.0)] 619 ), 620 ' is wrong!') 621 self.failUnless(numpy.alltrue(data['generatedtriangleattributelist'] == 622 [[tri_tag], [tri_tag]] 623 ), 624 ' is wrong!') 625 626 self.failUnless(numpy.alltrue(data['generatedsegmentlist'] == seglist), 627 ' is wrong!') 628 self.failUnless(numpy.alltrue(data['generatedsegmentmarkerlist'] == segattlist), 616 629 ' is wrong!') 617 630 # I copied these answers from the output, so bad test.. 618 self.failUnless(data['generatedtriangleneighborlist'] == \ 619 [(-1, 1, -1), (-1, 0, -1)], 631 self.failUnless(numpy.alltrue(data['generatedtriangleneighborlist'] == 632 [(-1, 1, -1), (-1, 0, -1)] 633 ), 620 634 ' is wrong!') 621 635 … … 635 649 data = generate_mesh(points,seglist,holelist,regionlist, 636 650 pointattlist,segattlist, mode, points) 637 self.failUnless(data['generatedpointattributelist'] == [[0.0],[0.0], 638 [10],[10]], 651 self.failUnless(numpy.alltrue(data['generatedpointattributelist'] == 652 [[0.0],[0.0],[10],[10]] 653 ), 639 654 ' is wrong!') 640 655 … … 644 659 data = generate_mesh(points,seglist,holelist,regionlist, 645 660 pointattlist,segattlist, mode, points) 646 self.failUnless( data['generatedpointattributelist'] == [[0.0],[0.0],647 [10],[10]],661 self.failUnless(numpy.alltrue(data['generatedpointattributelist'] == 662 [[0.0],[0.0],[10],[10]]), 648 663 ' is wrong!') 649 664 pointattlist = [[0.,1],[0.,1],[10.,20],[10.,20]] … … 652 667 pointattlist,segattlist, mode, points) 653 668 #print "data", data 654 self.failUnless( data['generatedpointattributelist'] == pointattlist,669 self.failUnless(numpy.alltrue(data['generatedpointattributelist'] == pointattlist), 655 670 ' is wrong!') 656 671 -
anuga_core/source_numpy_conversion/anuga/pmesh/test_mesh.py
r5912 r5951 426 426 vert = m.getMeshVerticeAttributes() 427 427 428 self.failUnless( vert[0] == [12.0, 2.0]and429 vert[1] == [9.0, 7.0]and430 vert[2] == [14.0,3.0]and431 vert[3] == [12.232233047033631, 4.4142135623730949]and432 vert[4] == [13.0, 2.5],428 self.failUnless(numpy.alltrue(vert[0] == [12.0, 2.0]) and 429 numpy.alltrue(vert[1] == [9.0, 7.0]) and 430 numpy.alltrue(vert[2] == [14.0,3.0]) and 431 numpy.alltrue(vert[3] == [12.232233047033631, 4.4142135623730949]) and 432 numpy.alltrue(vert[4] == [13.0, 2.5]), 433 433 'vertex attributes are wrong!') 434 434 … … 1653 1653 #print "dict['vertices']",dict['vertices'] 1654 1654 1655 self.failUnless( dict['vertices'] == answer,1656 1657 1658 self.failUnless( dict['vertices'] == verts,1659 1660 self.failUnless( dict['vertex_attributes'] == vert_as,1661 1662 1663 self.failUnless( dict['segments'][0] == [0,1],1655 self.failUnless(numpy.alltrue(dict['vertices'] == answer), 1656 'test_Mesh2IOTriangulationDict failed. test 2') 1657 1658 self.failUnless(numpy.alltrue(dict['vertices'] == verts), 1659 'test_Mesh2IOTriangulationDict failed. test vert') 1660 self.failUnless(numpy.alltrue(dict['vertex_attributes'] == vert_as), 1661 'test_Mesh2IOTriangulationDict failed. test vert ats') 1662 1663 self.failUnless(numpy.alltrue(dict['segments'][0] == [0,1]), 1664 1664 'test_Mesh2IODict failed. test 3') 1665 1665 1666 self.failUnless( dict['segment_tags'] == seg_tags,1666 self.failUnless(numpy.alltrue(dict['segment_tags'] == seg_tags), 1667 1667 'test_Mesh2IODict failed. test 3') 1668 1668 #print " dict['triangles'][0]", dict['triangles'][0] 1669 self.failUnless( dict['triangles'][0] == [3,2,4],1669 self.failUnless(numpy.alltrue(dict['triangles'][0] == [3,2,4]), 1670 1670 'test_Mesh2IODict failed. test 5') 1671 self.failUnless( dict['triangle_neighbors'][0] == [-1,2,3],1671 self.failUnless(numpy.alltrue(dict['triangle_neighbors'][0] == [-1,2,3]), 1672 1672 'test_Mesh2IODict failed. test 6') 1673 1673 #print "dict['triangle_tags'][0]", dict['triangle_tags'][0] … … 1714 1714 #print "dict['vertices']",dict['vertices'] 1715 1715 1716 self.failUnless( dict['vertices'] == answer,1716 self.failUnless(numpy.alltrue(dict['vertices'] == answer), 1717 1717 'test_Mesh2IOTriangulationDict failed. test 2') 1718 1718 1719 self.failUnless( dict['vertices'] == verts,1719 self.failUnless(numpy.alltrue(dict['vertices'] == verts), 1720 1720 'test_Mesh2IOTriangulationDict failed. test vert') 1721 self.failUnless( dict['vertex_attributes'] == vert_as,1721 self.failUnless(numpy.alltrue(dict['vertex_attributes'] == vert_as), 1722 1722 'test_Mesh2IOTriangulationDict failed. test vert ats') 1723 1723 1724 self.failUnless( dict['segments'][0] == [0,1],1724 self.failUnless(numpy.alltrue(dict['segments'][0] == [0,1]), 1725 1725 'test_Mesh2IODict failed. test 3') 1726 1726 1727 self.failUnless( dict['segment_tags'] == seg_tags,1727 self.failUnless(numpy.alltrue(dict['segment_tags'] == seg_tags), 1728 1728 'test_Mesh2IODict failed. test 3') 1729 1729 #print " dict['triangles'][0]", dict['triangles'][0] 1730 self.failUnless( dict['triangles'][0] == [3,2,4],1730 self.failUnless(numpy.alltrue(dict['triangles'][0] == [3,2,4]), 1731 1731 'test_Mesh2IODict failed. test 5') 1732 self.failUnless( dict['triangle_neighbors'][0] == [-1,2,3],1732 self.failUnless(numpy.alltrue(dict['triangle_neighbors'][0] == [-1,2,3]), 1733 1733 'test_Mesh2IODict failed. test 6') 1734 1734 #print "dict['triangle_tags'][0]", dict['triangle_tags'][0] 1735 self.failUnless( 1736 1735 self.failUnless(dict['triangle_tags'][0] == "1.3", 1736 'test_Mesh2IODict failed. test 7') 1737 1737 1738 1738 seg = m.getUserSegments() … … 2178 2178 'FAILED!') 2179 2179 vert= m.get_user_vertices(absolute=True) 2180 self.failUnless(vert==points_ab, 2181 'FAILED!') 2180 self.failUnless(numpy.alltrue(vert==points_ab), 'FAILED!') 2182 2181 2183 2182 … … 2194 2193 'FAILED!') 2195 2194 vert= m.get_user_vertices(absolute=True) 2196 self.failUnless(vert==points, 2197 'FAILED!') 2195 self.failUnless(numpy.alltrue(vert==points), 'FAILED!') 2198 2196 2199 2197 def test_add_verticesII(self): … … 2211 2209 'FAILED!') 2212 2210 vert= m.get_user_vertices(absolute=True) 2213 self.failUnless( vert==points_ab,2211 self.failUnless(numpy.alltrue(vert==points_ab), 2214 2212 'FAILED!') 2215 2213 … … 2227 2225 'FAILED!') 2228 2226 vert= m.get_user_vertices(absolute=True) 2229 self.failUnless(vert==points_ab, 2230 'FAILED!') 2227 self.failUnless(numpy.alltrue(vert==points_ab), 'FAILED!') 2231 2228 2232 2229 #geo = Geo_reference(56,23,21) … … 2248 2245 'FAILED!') 2249 2246 vert= m.get_user_vertices(absolute=True) 2250 self.failUnless( vert==points_ab,2247 self.failUnless(numpy.alltrue(vert==points_ab), 2251 2248 'FAILED!') 2252 2249 vert= m.get_user_vertices(absolute=False) -
anuga_core/source_numpy_conversion/anuga/shallow_water/data_manager.py
r5947 r5951 602 602 from Scientific.IO.NetCDF import NetCDFFile 603 603 604 self.precision = numpy.float #Use full precision604 self.precision = Float #Use full precision 605 605 606 606 Data_format.__init__(self, domain, 'sww', mode) … … 1080 1080 1081 1081 M = size #Number of lines 1082 xx = numpy.zeros((M,3), numpy.float)1083 yy = numpy.zeros((M,3), numpy.float)1084 zz = numpy.zeros((M,3), numpy.float)1082 xx = numpy.zeros((M,3), Float) 1083 yy = numpy.zeros((M,3), Float) 1084 zz = numpy.zeros((M,3), Float) 1085 1085 1086 1086 for i in range(M): … … 1126 1126 1127 1127 M = len(lines) #Number of lines 1128 x = numpy.zeros((M,3), numpy.float)1129 y = numpy.zeros((M,3), numpy.float)1130 z = numpy.zeros((M,3), numpy.float)1128 x = numpy.zeros((M,3), Float) 1129 y = numpy.zeros((M,3), Float) 1130 z = numpy.zeros((M,3), Float) 1131 1131 1132 1132 ##i = 0 … … 1443 1443 newcols = lenv # ncols_in_bounding_box 1444 1444 1445 telev = numpy.zeros(newcols, numpy.float)1446 tpoints = numpy.zeros((newcols, 2), numpy.float)1445 telev = numpy.zeros(newcols, Float) 1446 tpoints = numpy.zeros((newcols, 2), Float) 1447 1447 1448 1448 local_index = 0 … … 1993 1993 #the temporal dimension 1994 1994 if verbose: print 'Reducing quantity %s' %quantity 1995 q_reduced = numpy.zeros( number_of_points, numpy.float )1995 q_reduced = numpy.zeros( number_of_points, Float ) 1996 1996 1997 1997 if timestep is not None: … … 2059 2059 assert len(vertex_points.shape) == 2 2060 2060 2061 grid_points = numpy.zeros ( (ncols*nrows, 2), numpy.float )2061 grid_points = numpy.zeros ( (ncols*nrows, 2), Float ) 2062 2062 2063 2063 … … 2358 2358 # the temporal dimension 2359 2359 if verbose: print 'Reducing quantity %s' %quantity 2360 q_reduced = numpy.zeros( number_of_points, numpy.float )2360 q_reduced = numpy.zeros( number_of_points, Float ) 2361 2361 2362 2362 for k in range(number_of_points): … … 2667 2667 from Scientific.IO.NetCDF import NetCDFFile 2668 2668 2669 precision = numpy.float2669 precision = Float 2670 2670 2671 2671 msg = 'Must use latitudes and longitudes for minlat, maxlon etc' … … 2821 2821 #Cleanup 2822 2822 missing = (amplitudes == nan_ha) 2823 if sometrue(missing):2823 if numpy.sometrue(missing): 2824 2824 if fail_on_NaN: 2825 2825 msg = 'NetCDFFile %s contains missing values'\ … … 2925 2925 sww.store_header(outfile, times, number_of_volumes, 2926 2926 number_of_points, description=description, 2927 verbose=verbose,sww_precision= numpy.float)2927 verbose=verbose,sww_precision=Float) 2928 2928 2929 2929 # Store 2930 2930 from anuga.coordinate_transforms.redfearn import redfearn 2931 x = numpy.zeros(number_of_points, numpy.float) #Easting2932 y = numpy.zeros(number_of_points, numpy.float) #Northing2931 x = numpy.zeros(number_of_points, Float) #Easting 2932 y = numpy.zeros(number_of_points, Float) #Northing 2933 2933 2934 2934 … … 2987 2987 outfile.variables['z'][:] = z #FIXME HACK for bacwards compat. 2988 2988 outfile.variables['elevation'][:] = z 2989 outfile.variables['volumes'][:] = volumes.astype( numpy.int32) #For Opteron 642989 outfile.variables['volumes'][:] = volumes.astype(Int) #For Opteron 64 2990 2990 2991 2991 … … 3123 3123 d = len(q) 3124 3124 3125 T = numpy.zeros(N, numpy.float) #Time3126 Q = numpy.zeros((N, d), numpy.float) #Values3125 T = numpy.zeros(N, Float) #Time 3126 Q = numpy.zeros((N, d), Float) #Values 3127 3127 3128 3128 for i, line in enumerate(lines): … … 3570 3570 if verbose: print 'Processing row %d of %d' %(i, nrows_new) 3571 3571 lower_index = global_index 3572 telev = numpy.zeros(ncols_new, numpy.float)3572 telev = numpy.zeros(ncols_new, Float) 3573 3573 local_index = 0 3574 3574 trow = i * cellsize_ratio … … 3668 3668 from anuga.coordinate_transforms.redfearn import redfearn 3669 3669 3670 precision = numpy.float # So if we want to change the precision its done here3670 precision = Float # So if we want to change the precision its done here 3671 3671 3672 3672 # go in to the bath dir and load the only file, … … 3795 3795 #Store 3796 3796 from anuga.coordinate_transforms.redfearn import redfearn 3797 x = numpy.zeros(number_of_points, numpy.float) #Easting3798 y = numpy.zeros(number_of_points, numpy.float) #Northing3797 x = numpy.zeros(number_of_points, Float) #Easting 3798 y = numpy.zeros(number_of_points, Float) #Northing 3799 3799 3800 3800 if verbose: print 'Making triangular grid' … … 3859 3859 outfile.variables['z'][:] = z # FIXME (Ole): Remove once viewer has been recompiled and changed to use elevation instead of z 3860 3860 outfile.variables['elevation'][:] = z 3861 outfile.variables['volumes'][:] = volumes.astype( numpy.int32) # On Opteron 643861 outfile.variables['volumes'][:] = volumes.astype(Int) # On Opteron 64 3862 3862 3863 3863 stage = outfile.variables['stage'] … … 4035 4035 lat_name = 'LAT' 4036 4036 time_name = 'TIME' 4037 precision = numpy.float # So if we want to change the precision its done here4037 precision = Float # So if we want to change the precision its done here 4038 4038 class Write_nc: 4039 4039 """ … … 4260 4260 lonlatdep = p_array.array('f') ###NumPy? 4261 4261 lonlatdep.read(mux_file, columns * points_num) 4262 lonlatdep = numpy.array(lonlatdep, typecode=numpy.float)4262 lonlatdep = numpy.array(lonlatdep, dtype=Float) 4263 4263 lonlatdep = numpy.reshape(lonlatdep, (points_num, columns)) 4264 4264 … … 4267 4267 lon_sorted.sort() 4268 4268 4269 if not lon == lon_sorted:4269 if not numpy.alltrue(lon == lon_sorted): 4270 4270 msg = "Longitudes in mux file are not in ascending order" 4271 4271 raise IOError, msg … … 4273 4273 lat_sorted.sort() 4274 4274 4275 if not lat == lat_sorted:4275 if not numpy.alltrue(lat == lat_sorted): 4276 4276 msg = "Latitudes in mux file are not in ascending order" 4277 4277 … … 4287 4287 hz_p_array = p_array.array('f') 4288 4288 hz_p_array.read(mux_file, points_num) 4289 hz_p = numpy.array(hz_p_array, typecode=numpy.float)4289 hz_p = numpy.array(hz_p_array, dtype=Float) 4290 4290 hz_p = numpy.reshape(hz_p, (len(lon), len(lat))) 4291 4291 hz_p = numpy.transpose(hz_p) #mux has lat varying fastest, nc has long v.f. … … 4365 4365 QUANTITY = 2 4366 4366 4367 long_lat_dep = ensure_numeric(long_lat_dep, numpy.float)4367 long_lat_dep = ensure_numeric(long_lat_dep, Float) 4368 4368 4369 4369 num_points = long_lat_dep.shape[0] … … 4413 4413 # use numeric transpose, after reshaping the quantity vector 4414 4414 # quantity = zeros(len(long_lat_dep), Float) 4415 quantity = numpy.zeros(num_points, numpy.float)4415 quantity = numpy.zeros(num_points, Float) 4416 4416 4417 4417 # print 'num',num_points … … 4795 4795 4796 4796 points_utm=ensure_numeric(points_utm) 4797 assert ensure_numeric(mesh_dic['generatedpointlist']) == \4798 ensure_numeric(points_utm)4797 assert numpy.alltrue(ensure_numeric(mesh_dic['generatedpointlist']) == 4798 ensure_numeric(points_utm)) 4799 4799 4800 4800 volumes = mesh_dic['generatedtrianglelist'] … … 4812 4812 sww = Write_sww() 4813 4813 sww.store_header(outfile, times, len(volumes), len(points_utm), 4814 verbose=verbose,sww_precision= numpy.float)4814 verbose=verbose,sww_precision=Float) 4815 4815 outfile.mean_stage = mean_stage 4816 4816 outfile.zscale = zscale … … 4835 4835 xmomentum=xmomentum, 4836 4836 ymomentum=ymomentum, 4837 sww_precision= numpy.float)4837 sww_precision=Float) 4838 4838 j += 1 4839 4839 if verbose: sww.verbose_quantities(outfile) … … 4871 4871 numSrc=len(filenames) 4872 4872 4873 file_params=-1*numpy.ones(3, numpy.float) #[nsta,dt,nt]4873 file_params=-1*numpy.ones(3,Float) #[nsta,dt,nt] 4874 4874 4875 4875 # Convert verbose to int C flag … … 4880 4880 4881 4881 if permutation is None: 4882 permutation = ensure_numeric([], numpy.float)4882 permutation = ensure_numeric([], Float) 4883 4883 4884 4884 # Call underlying C implementation urs2sts_ext.c … … 4930 4930 4931 4931 times=dt*numpy.arange(parameters_index) 4932 latitudes=numpy.zeros(number_of_selected_stations, numpy.float)4933 longitudes=numpy.zeros(number_of_selected_stations, numpy.float)4934 elevation=numpy.zeros(number_of_selected_stations, numpy.float)4935 quantity=numpy.zeros((number_of_selected_stations, parameters_index), numpy.float)4932 latitudes=numpy.zeros(number_of_selected_stations, Float) 4933 longitudes=numpy.zeros(number_of_selected_stations, Float) 4934 elevation=numpy.zeros(number_of_selected_stations, Float) 4935 quantity=numpy.zeros((number_of_selected_stations, parameters_index), Float) 4936 4936 4937 4937 … … 5040 5040 if weights is None: 5041 5041 # Default is equal weighting 5042 weights=numpy.ones(numSrc, numpy.float)/numSrc5042 weights=numpy.ones(numSrc,Float)/numSrc 5043 5043 else: 5044 5044 weights = ensure_numeric(weights) … … 5174 5174 # 0 to number_of_points-1 5175 5175 if permutation is None: 5176 permutation = numpy.arange(number_of_points, typecode=numpy.int)5176 permutation = numpy.arange(number_of_points, dtype=Int) 5177 5177 5178 5178 … … 5191 5191 description=description, 5192 5192 verbose=verbose, 5193 sts_precision= numpy.float)5193 sts_precision=Float) 5194 5194 5195 5195 # Store 5196 5196 from anuga.coordinate_transforms.redfearn import redfearn 5197 x = numpy.zeros(number_of_points, numpy.float) # Easting5198 y = numpy.zeros(number_of_points, numpy.float) # Northing5197 x = numpy.zeros(number_of_points, Float) # Easting 5198 y = numpy.zeros(number_of_points, Float) # Northing 5199 5199 5200 5200 # Check zone boundaries … … 5229 5229 5230 5230 elevation = numpy.resize(elevation,outfile.variables['elevation'][:].shape) 5231 outfile.variables['permutation'][:] = permutation.astype( numpy.int32) # On Opteron 645231 outfile.variables['permutation'][:] = permutation.astype(Int) # On Opteron 64 5232 5232 outfile.variables['x'][:] = x - geo_ref.get_xllcorner() 5233 5233 outfile.variables['y'][:] = y - geo_ref.get_yllcorner() … … 5531 5531 outfile.variables['z'][:] = elevation 5532 5532 outfile.variables['elevation'][:] = elevation #FIXME HACK 5533 outfile.variables['volumes'][:] = volumes.astype( numpy.int32) #On Opteron 645533 outfile.variables['volumes'][:] = volumes.astype(Int) #On Opteron 64 5534 5534 5535 5535 q = 'elevation' … … 5539 5539 5540 5540 5541 def store_quantities(self, outfile, sww_precision= numpy.float32,5541 def store_quantities(self, outfile, sww_precision=Float32, 5542 5542 slice_index=None, time=None, 5543 5543 verbose=False, **quant): … … 5706 5706 number_of_points, 5707 5707 description='Converted from URS mux2 format', 5708 sts_precision= numpy.float32,5708 sts_precision=Float32, 5709 5709 verbose=False): 5710 5710 """ … … 5765 5765 5766 5766 # Doing sts_precision instead of Float gives cast errors. 5767 outfile.createVariable('time', Float r, ('number_of_timesteps',))5767 outfile.createVariable('time', Float, ('number_of_timesteps',)) 5768 5768 5769 5769 for q in Write_sts.sts_quantities: … … 5874 5874 outfile.variables[q+Write_sts.RANGE][1] = max(elevation) 5875 5875 5876 def store_quantities(self, outfile, sts_precision= numpy.float32,5876 def store_quantities(self, outfile, sts_precision=Float32, 5877 5877 slice_index=None, time=None, 5878 5878 verbose=False, **quant): … … 5964 5964 lonlatdep = p_array.array('f') 5965 5965 lonlatdep.read(mux_file, columns * self.points_num) 5966 lonlatdep = numpy.array(lonlatdep, typecode=numpy.float)5966 lonlatdep = numpy.array(lonlatdep, dtype=Float) 5967 5967 lonlatdep = numpy.reshape(lonlatdep, (self.points_num, columns)) 5968 5968 #print 'lonlatdep',lonlatdep … … 5993 5993 hz_p_array = p_array.array('f') 5994 5994 hz_p_array.read(self.mux_file, self.points_num) 5995 hz_p = numpy.array(hz_p_array, typecode=numpy.float)5995 hz_p = numpy.array(hz_p_array, dtype=Float) 5996 5996 self.iter_time_step += 1 5997 5997 … … 6108 6108 #array to store data, number in there is to allow float... 6109 6109 #i'm sure there is a better way! 6110 data=numpy.array([], typecode=numpy.float)6110 data=numpy.array([],dtype=Float) 6111 6111 data=numpy.resize(data,((len(lines)-1),len(header_fields))) 6112 6112 # print 'number of fields',range(len(header_fields)) … … 6685 6685 6686 6686 # Get the relevant quantities (Convert from single precison) 6687 elevation = numpy.array(fid.variables['elevation'][:], numpy.float)6688 stage = numpy.array(fid.variables['stage'][:], numpy.float)6687 elevation = numpy.array(fid.variables['elevation'][:], Float) 6688 stage = numpy.array(fid.variables['stage'][:], Float) 6689 6689 6690 6690 -
anuga_core/source_numpy_conversion/anuga/shallow_water/test_data_manager.py
r5947 r5951 4964 4964 'failed') 4965 4965 4966 self.failUnless( m2d == [[5,6],[9,10]],4967 4966 self.failUnless(numpy.alltrue(m2d == [[5,6],[9,10]]), 4967 'failed') 4968 4968 4969 4969 def test_get_min_max_indexes_lat_ascending(self): … … 5030 5030 self.failUnless(latitudes_new == [-30, -35, -40] and \ 5031 5031 longitudes_new == [148, 149,150], 5032 5033 self.failUnless( m2d == [[0,1,2],[4,5,6],[8,9,10]],5034 5032 'failed') 5033 self.failUnless(numpy.alltrue(m2d == [[0,1,2],[4,5,6],[8,9,10]]), 5034 'failed') 5035 5035 5036 5036 def test_get_min_max_indexes3(self): -
anuga_core/source_numpy_conversion/anuga/shallow_water/tsunami_okada.py
r5901 r5951 31 31 32 32 import numpy 33 from anuga.config import Float 34 33 35 34 36 def earthquake_tsunami(ns,NSMAX,length, width, strike, depth,\ … … 44 46 zrec=zrec0.get_vertex_values(xy=True) 45 47 46 x0= numpy.zeros(ns, numpy.float)47 y0= numpy.zeros(ns, numpy.float)48 x0= numpy.zeros(ns,Float) 49 y0= numpy.zeros(ns,Float) 48 50 if ns ==1: 49 51 x0[0]=xi … … 150 152 zrec=self.zrec 151 153 #initialization 152 disp0=numpy.zeros(3, numpy.float)153 strain0=numpy.zeros(6, numpy.float)154 tilt0 = numpy.zeros(2, numpy.float)155 dislocations=numpy.zeros(ns, numpy.float)156 depths=numpy.zeros(ns, numpy.float)157 strikes= numpy.zeros(ns, numpy.float)158 lengths= numpy.zeros(ns, numpy.float)159 slips= numpy.zeros(ns, numpy.float)160 rakes= numpy.zeros(ns, numpy.float)161 widths= numpy.zeros(ns, numpy.float)162 dips= numpy.zeros(ns, numpy.float)163 strikes= numpy.zeros(ns, numpy.float)164 strikes= numpy.zeros(ns, numpy.float)165 strain = numpy.zeros((N,6), numpy.float)166 disp = numpy.zeros((N,3), numpy.float)167 tilt = numpy.zeros((N,2), numpy.float)168 xs =numpy.zeros(ns, numpy.float)169 ys =numpy.zeros(ns, numpy.float)154 disp0=numpy.zeros(3,Float) 155 strain0=numpy.zeros(6,Float) 156 tilt0 = numpy.zeros(2,Float) 157 dislocations=numpy.zeros(ns,Float) 158 depths=numpy.zeros(ns,Float) 159 strikes= numpy.zeros(ns,Float) 160 lengths= numpy.zeros(ns,Float) 161 slips= numpy.zeros(ns,Float) 162 rakes= numpy.zeros(ns,Float) 163 widths= numpy.zeros(ns,Float) 164 dips= numpy.zeros(ns,Float) 165 strikes= numpy.zeros(ns,Float) 166 strikes= numpy.zeros(ns,Float) 167 strain = numpy.zeros((N,6),Float) 168 disp = numpy.zeros((N,3),Float) 169 tilt = numpy.zeros((N,2),Float) 170 xs =numpy.zeros(ns,Float) 171 ys =numpy.zeros(ns,Float) 170 172 z=[] 171 173 if ns==1: … … 213 215 xrec=y 214 216 yrec=x 217 ## print 'xrec=%s' % str(xrec) 218 ## print 'yrec=%s' % str(yrec) 219 ## print 'zrec=%s' % str(zrec) 215 220 for i in range(0,len(zrec[0])): 216 if zrec[0][i]==yrec and zrec[1][i]==xrec: 221 if numpy.alltrue(zrec[0][i]==yrec) and \ 222 numpy.alltrue(zrec[1][i]==xrec): 223 ## print 'Setting Z!' 217 224 Z=zrec[2][i] 218 225 Z=0.001*Z … … 377 384 378 385 F0=0.0 379 U=numpy.zeros((12,1), numpy.float)380 DUA=numpy.zeros((12,1), numpy.float)381 DUB=numpy.zeros((12,1), numpy.float)382 DUC=numpy.zeros((12,1), numpy.float)386 U=numpy.zeros((12,1),Float) 387 DUA=numpy.zeros((12,1),Float) 388 DUB=numpy.zeros((12,1),Float) 389 DUC=numpy.zeros((12,1),Float) 383 390 384 391 … … 523 530 WZ=self.WZ 524 531 525 DUA=numpy.zeros((12,1), numpy.float)526 DU=numpy.zeros((12,1), numpy.float)527 U=numpy.zeros((12,1), numpy.float)532 DUA=numpy.zeros((12,1),Float) 533 DU=numpy.zeros((12,1),Float) 534 U=numpy.zeros((12,1),Float) 528 535 #----- 529 536 for I in range(0,12): … … 633 640 # DATA PI2/6.283185307179586D0/ 634 641 635 DUB=numpy.zeros((12,1), numpy.float)636 DU=numpy.zeros((12,1), numpy.float)637 U=numpy.zeros((12,1), numpy.float)642 DUB=numpy.zeros((12,1),Float) 643 DU=numpy.zeros((12,1),Float) 644 U=numpy.zeros((12,1),Float) 638 645 639 646 F0=0.0 … … 807 814 # DATA PI2/6.283185307179586D0/ 808 815 809 DUC=numpy.zeros((12,1), numpy.float)810 DU=numpy.zeros((12,1), numpy.float)811 U=numpy.zeros((12,1), numpy.float)816 DUC=numpy.zeros((12,1),Float) 817 DU=numpy.zeros((12,1),Float) 818 U=numpy.zeros((12,1),Float) 812 819 813 820 F0=0.0 … … 1001 1008 EPS=0.000001 1002 1009 1003 XI=numpy.zeros(2, numpy.float)1004 ET=numpy.zeros(2, numpy.float)1005 KXI=numpy.zeros(2, numpy.float)1006 KET=numpy.zeros(2, numpy.float)1007 U=numpy.zeros(12, numpy.float)1008 DU=numpy.zeros(12, numpy.float)1009 DUA=numpy.zeros(12, numpy.float)1010 DUB=numpy.zeros(12, numpy.float)1011 DUC=numpy.zeros(12, numpy.float)1010 XI=numpy.zeros(2,Float) 1011 ET=numpy.zeros(2,Float) 1012 KXI=numpy.zeros(2,Float) 1013 KET=numpy.zeros(2,Float) 1014 U=numpy.zeros(12,Float) 1015 DU=numpy.zeros(12,Float) 1016 DUA=numpy.zeros(12,Float) 1017 DUB=numpy.zeros(12,Float) 1018 DUC=numpy.zeros(12,Float) 1012 1019 1013 1020 #----- … … 1206 1213 # DATA F0,F2,PI2/0.D0,2.D0,6.283185307179586D0/ 1207 1214 1208 U=numpy.zeros(12, numpy.float)1209 DU=numpy.zeros(12, numpy.float)1210 DUA=numpy.zeros(12, numpy.float)1215 U=numpy.zeros(12,Float) 1216 DU=numpy.zeros(12,Float) 1217 DUA=numpy.zeros(12,Float) 1211 1218 F0 =0.0 1212 1219 F2=2.0 … … 1337 1344 # DATA F0,F1,F2,PI2/0.D0,1.D0,2.D0,6.283185307179586D0/ 1338 1345 1339 DUB=numpy.zeros(12, numpy.float)1340 DU=numpy.zeros(12, numpy.float)1341 U=numpy.zeros(12, numpy.float)1346 DUB=numpy.zeros(12,Float) 1347 DU=numpy.zeros(12,Float) 1348 U=numpy.zeros(12,Float) 1342 1349 1343 1350 F0=0.0 … … 1502 1509 # DATA F0,F1,F2,F3,PI2/0.D0,1.D0,2.D0,3.D0,6.283185307179586D0/ 1503 1510 1504 DUC=numpy.zeros(12, numpy.float)1505 DU=numpy.zeros(12, numpy.float)1506 U=numpy.zeros(12, numpy.float)1511 DUC=numpy.zeros(12,Float) 1512 DU=numpy.zeros(12,Float) 1513 U=numpy.zeros(12,Float) 1507 1514 1508 1515 F0=0.0
Note: See TracChangeset
for help on using the changeset viewer.