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')