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