def setUp(self):
self.filename = tempfile.mktemp('.csv')
self.titles = ['Time', 'Lat', 'Lon', 'Alt']
self.data = [['1', '2', '3'], [0, 1, 2], [1, 0, -1], [0.3, 1.5, 1.7]]
self.times = ['1', '2', '3']
self.lats = [0, 1, 2]
self.lons = [1, 0, -1]
self.alts = [0.3, 1.5, 1.7]
self.data_as_str = [['1', '2', '3'], ['0', '1', '2'], ['1', '0', '-1'],
['0.3', '1.5', '1.7']]
self.format = ['s', 'i', 'i', 'f']
f = einput.EgadsCsv(self.filename, 'w')
f.write(self.titles)
f.writerows(self.data)
f.close()
self.f = einput.EgadsCsv(self.filename)
def setUp(self):
self.filename = tempfile.mktemp('.csv')
self.titles = 'Time, Lat, Lon, Alt'
self.data = [['1', 0, 1, 0.3], ['2', 1, 0, 1.5], ['3', 2, -1, 1.7]]
self.data_as_str = [['1', '0', '1', '0.3'], ['2', '1', '0', '1.5'],
['3', '2', '-1', '1.7']]
self.format = ['s', 'i', 'i', 'f']
f = input.EgadsCsv(self.filename, 'w')
f.write(self.titles)
f.writerows(self.data)
f.close()
def test_convert_nc_to_csv_egadsnetcdf(self):
""" Test conversion of NetCDF to Nasa/Ames CSV, using the EgadsNetCdf class """
f = einput.EgadsNetCdf(self.ncfilename)
f.convert_to_csv(self.csvfilename)
f.close()
g = einput.EgadsCsv()
g.open(self.csvfilename, 'r')
lines = g.read()
author = lines[1][0]
computer = lines[3][0]
raw = lines[-5:]
time = []
data = []
for i in raw:
time.append(float(i[0]))
data.append(float(i[1]))
self.assertEqual('John Doe ([email protected])', author,
'Originator values do not match')
self.assertEqual('computer', computer, 'Source values do not match')
self.assertListEqual(self.data1.value.tolist(), data,
'data and data1 values do not match')
self.assertListEqual(self.data2.value.tolist(), time,
'time and data2 values do not match')