def test_export_csv(self): "Test export_csv function" file_name = 'test' path = '/var/lib/cuadrnt' headers = ('foo', 'bar') data = [('Bjorn', 'Barrefors')] export_file = '%s/%s.csv' % (path, file_name) export_csv(file_name=file_name, path=path, headers=headers, data=data) expected = 'foo,bar\nBjorn,Barrefors\n' fs = open(export_file, 'r') result = fs.read() self.assertEqual(result, expected) os.remove(export_file)
def test_export_csv(self): "Test export_csv function" file_name = 'test' path = '/var/lib/cuadrnt' headers = ('foo', 'bar') data = [('Bjorn', 'Barrefors')] export_file = '%s/%s.csv' % (path, file_name) export_csv(headers=headers, data=data, path=path, file_name=file_name) expected = 'foo,bar\nBjorn,Barrefors\n' fd = open(export_file, 'r') result = fd.read() fd.close() self.assertEqual(result, expected) os.remove(export_file)
def export_data(self, dataset_name): """ Get data from DB and export to file for usage in visualization """ # get data from DB coll = 'dataset_popularity' pipeline = list() match = {'$match':{'name':dataset_name}} pipeline.append(match) db_data = self.storage.get_data(coll=coll, pipeline=pipeline) headers = ('dataset_name', 'date', 'popularity') data = list() for data_entry in db_data: data.append(tuple(data_entry['name'], data_entry['date'], data_entry['n_accesses']*data_entry['n_cpus']*data_entry['n_users'])) export_csv(headers=headers, data=data, file_name='single_dataset')