Changeset 7112
- Timestamp:
- May 28, 2009, 9:46:03 AM (16 years ago)
- File:
-
- 1 edited
Legend:
- Unmodified
- Added
- Removed
-
anuga_core/source/anuga/utilities/test_csv_tools.py
r7104 r7112 337 337 338 338 339 def test_latex_example(self): 340 """Test merging two CSV files - example from latex doc.""" 341 342 fd = open('alpha.csv', 'w') 343 csv_fd = csv.writer(fd) 344 csv_fd.writerow(['time', 'hours', 'stage', 'depth']) 345 csv_fd.writerow(['3600', '1.00', '100.3', '10.2']) 346 csv_fd.writerow(['3636', '1.01', '100.3', '10.0']) 347 csv_fd.writerow(['3672', '1.02', '100.3', '9.7']) 348 csv_fd.writerow(['3708', '1.03', '100.3', '8.9']) 349 csv_fd.writerow(['3744', '1.04', '100.3', '7.1']) 350 fd.close() 351 352 fd = open('beta.csv', 'w') 353 csv_fd = csv.writer(fd) 354 csv_fd.writerow(['time', 'hours', 'stage', 'depth']) 355 csv_fd.writerow(['3600', '1.00', '100.3', '11.3']) 356 csv_fd.writerow(['3636', '1.01', '100.3', '10.5']) 357 csv_fd.writerow(['3672', '1.02', '100.3', '10.0']) 358 csv_fd.writerow(['3708', '1.03', '100.3', '9.7']) 359 csv_fd.writerow(['3744', '1.04', '100.3', '8.2']) 360 fd.close() 361 362 file_title_list = [('alpha.csv', 'alpha'), 363 ('beta.csv', 'beta')] 364 csv_tools.merge_csv_key_values(file_title_list, 365 'gamma.csv', 366 key_col='hours', 367 data_col='depth') 368 369 expected = '''hours,alpha,beta 370 1.00,10.2,11.3 371 1.01,10.0,10.5 372 1.02,9.7,10.0 373 1.03,8.9,9.7 374 1.04,7.1,8.2 375 ''' 376 377 got = self.get_file_contents('gamma.csv') 378 msg = ('Merging two files,\n' 379 'expected file=\n' 380 '--------------------\n' 381 '%s' 382 '--------------------\n' 383 'got file=\n' 384 '--------------------\n' 385 '%s' 386 '--------------------\n' 387 % (expected, got)) 388 self.failUnless(self.str_cmp(got, expected), msg) 389 390 try: 391 os.remove('alpha.csv') 392 os.remove('beta.csv') 393 os.remove('gamma.csv') 394 except: 395 pass 396 397 339 398 def str_cmp(self, str1, str2): 340 399 '''Compare 2 strings, removing end-of-line stuff first.'''
Note: See TracChangeset
for help on using the changeset viewer.