- Timestamp:
- Jun 3, 2010, 12:45:54 PM (13 years ago)
- File:
-
- 1 edited
Legend:
- Unmodified
- Added
- Removed
-
trunk/anuga_core/source/anuga/shallow_water/test_data_manager.py
r7771 r7772 16 16 from Scientific.IO.NetCDF import NetCDFFile 17 17 18 from anuga.anuga_exceptions import ANUGAError 18 19 from anuga.shallow_water.data_manager import * 19 20 from anuga.shallow_water.sww_file import SWW_file … … 23 24 from anuga.utilities.system_tools import get_pathname_from_package 24 25 from anuga.utilities.file_utils import del_dir 25 from anuga.file.csv_file import load_csv_as_dict, load_csv_as_array26 from anuga.anuga_exceptions import ANUGAError27 26 from anuga.utilities.numerical_tools import ensure_numeric, mean 28 27 from anuga.config import netcdf_mode_r, netcdf_mode_w, netcdf_mode_a 29 28 from anuga.config import netcdf_float, epsilon, g 29 30 from anuga.file.csv_file import load_csv_as_dict, load_csv_as_array 31 from anuga.file.sts import create_sts_boundary 30 32 31 33 … … 2252 2254 new_origin)),points_utm) 2253 2255 os.remove(filename) 2254 2255 def test_get_data_from_file(self):2256 # from anuga.abstract_2d_finite_volumes.util import get_data_from_file2257 2258 import os2259 2260 fileName = tempfile.mktemp(".txt")2261 # print"filename",fileName2262 file = open(fileName,"w")2263 file.write("elevation, stage\n\2264 1.0, 3 \n\2265 0.0, 4 \n\2266 4.0, 3 \n\2267 1.0, 6 \n")2268 file.close()2269 2270 header,x = get_data_from_file(fileName)2271 # print 'x',x2272 os.remove(fileName)2273 2274 assert num.allclose(x[:,0], [1.0, 0.0,4.0, 1.0])2275 2276 def test_get_data_from_file1(self):2277 # from anuga.abstract_2d_finite_volumes.util import get_data_from_file2278 2279 import os2280 2281 fileName = tempfile.mktemp(".txt")2282 # print"filename",fileName2283 file = open(fileName,"w")2284 file.write("elevation stage\n\2285 1.3 3 \n\2286 0.0 4 \n\2287 4.5 3.5 \n\2288 1.0 6 \n")2289 file.close()2290 2291 header, x = get_data_from_file(fileName,separator_value=' ')2292 os.remove(fileName)2293 # x = get_data_from_file(fileName)2294 # print '1x',x[:,0]2295 2296 assert num.allclose(x[:,0], [1.3, 0.0,4.5, 1.0])2297 2298 def test_store_parameters(self):2299 """tests store temporary file2300 """2301 2302 from os import sep, getenv2303 2304 output_dir=''2305 file_name='details.csv'2306 2307 kwargs = {'file_name':'new2.txt',2308 'output_dir':output_dir,2309 'file_name':file_name,2310 'who':'me',2311 'what':'detail',2312 'how':2,2313 'why':241,2314 # 'completed':3452315 }2316 store_parameters(verbose=False,**kwargs)2317 2318 temp='detail_temp.csv'2319 fid = open(temp)2320 file_header = fid.readline()2321 file_line = fid.readline()2322 fid.close()2323 2324 2325 keys = kwargs.keys()2326 keys.sort()2327 line=''2328 header=''2329 count=02330 #used the sorted keys to create the header and line data2331 for k in keys:2332 # print "%s = %s" %(k, kwargs[k])2333 header = header+str(k)2334 line = line+str(kwargs[k])2335 count+=12336 if count <len(kwargs):2337 header = header+','2338 line = line+','2339 header+='\n'2340 line+='\n'2341 2342 2343 #file exists2344 assert access(temp,F_OK)2345 assert header == file_header2346 assert line == file_line2347 2348 os.remove(temp)2349 2350 def test_store_parameters1(self):2351 """tests store in temporary file and other file2352 """2353 2354 from os import sep, getenv2355 2356 output_dir=''2357 file_name='details.csv'2358 2359 kwargs = {'file_name':'new2.txt',2360 'output_dir':output_dir,2361 'file_name':file_name,2362 'who':'me',2363 'what':'detail',2364 'how':2,2365 'why':241,2366 # 'completed':3452367 }2368 store_parameters(verbose=False,**kwargs)2369 2370 kwargs['how']=552371 kwargs['completed']=3452372 2373 keys = kwargs.keys()2374 keys.sort()2375 line=''2376 header=''2377 count=02378 #used the sorted keys to create the header and line data2379 for k in keys:2380 # print "%s = %s" %(k, kwargs[k])2381 header = header+str(k)2382 line = line+str(kwargs[k])2383 count+=12384 if count <len(kwargs):2385 header = header+','2386 line = line+','2387 header+='\n'2388 line+='\n'2389 2390 kwargs['how']=552391 kwargs['completed']=3452392 2393 store_parameters(verbose=False,**kwargs)2394 2395 # temp='detail_temp.csv'2396 fid = open(file_name)2397 file_header = fid.readline()2398 file_line1 = fid.readline()2399 file_line2 = fid.readline()2400 fid.close()2401 2402 2403 #file exists2404 # print 'header',header,'line',line2405 # print 'file_header',file_header,'file_line1',file_line1,'file_line2',file_line22406 assert access(file_name,F_OK)2407 assert header == file_header2408 assert line == file_line12409 2410 temp='detail_temp.csv'2411 os.remove(temp)2412 os.remove(file_name)2413 2414 def test_store_parameters2(self):2415 """tests appending the data to the end of an existing file2416 """2417 2418 from os import sep, getenv2419 2420 output_dir=''2421 file_name='details.csv'2422 2423 kwargs = {'file_name':'new2.txt',2424 'output_dir':output_dir,2425 'file_name':file_name,2426 'who':'me',2427 'what':'detail',2428 'how':2,2429 'why':241,2430 'completed':3452431 }2432 store_parameters(verbose=False,**kwargs)2433 2434 kwargs['how']=552435 kwargs['completed']=23.545322436 2437 store_parameters(verbose=False,**kwargs)2438 2439 keys = kwargs.keys()2440 keys.sort()2441 line=''2442 header=''2443 count=02444 #used the sorted keys to create the header and line data2445 for k in keys:2446 # print "%s = %s" %(k, kwargs[k])2447 header = header+str(k)2448 line = line+str(kwargs[k])2449 count+=12450 if count <len(kwargs):2451 header = header+','2452 line = line+','2453 header+='\n'2454 line+='\n'2455 2456 fid = open(file_name)2457 file_header = fid.readline()2458 file_line1 = fid.readline()2459 file_line2 = fid.readline()2460 fid.close()2461 2462 assert access(file_name,F_OK)2463 assert header == file_header2464 assert line == file_line22465 2466 os.remove(file_name)2467 2256 2468 2257
Note: See TracChangeset
for help on using the changeset viewer.