Changeset 7112


Ignore:
Timestamp:
May 28, 2009, 9:46:03 AM (11 years ago)
Author:
rwilson
Message:

Added Latex example code test.

File:
1 edited

Legend:

Unmodified
Added
Removed
  • anuga_core/source/anuga/utilities/test_csv_tools.py

    r7104 r7112  
    337337
    338338
     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
     3701.00,10.2,11.3
     3711.01,10.0,10.5
     3721.02,9.7,10.0
     3731.03,8.9,9.7
     3741.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
    339398    def str_cmp(self, str1, str2):
    340399        '''Compare 2 strings, removing end-of-line stuff first.'''
Note: See TracChangeset for help on using the changeset viewer.