def test_print_array(self):
folder_path = 'smallData'
resolution = 5
output_file = 'tmp.csv'
key_file = 'cgm_small.csv'
# pull all the folders in the file
files_lst = [
f for f in listdir(folder_path) if isfile(join(folder_path, f))
]
# import all the files into a list of ImportData objects
data_lst = []
for f in files_lst:
data_lst.append(data_import.ImportData(folder_path + '/' + f))
# convert to zipped, rounded data
zip_data = []
for data in data_lst:
zip_data.append(data_import.roundTimeArray(data, resolution))
# print to a csv file
data_import.printArray(zip_data, files_lst, output_file, key_file)
self.assertTrue(os.path.exists(output_file))
os.remove(output_file)
def test_printarray_bolus_cgm(self):
bolus_data = data_import.ImportData('smallData/bolus_small.csv')
cgm_data = data_import.ImportData('smallData/cgm_small.csv')
cgm_zip = data_import.roundTimeArray(cgm_data, 60, 'average')
bolus_zip = data_import.roundTimeArray(bolus_data, 60, 'sum')
data_list = [cgm_zip, bolus_zip]
ann_list = ['cgm_small', 'bolus_small']
base_name = 'test_printarray.csv'
key_file = 'cgm_small'
data_import.printArray(data_list, ann_list, base_name, key_file)
assert os.path.exists(base_name)
with open(base_name, 'r') as f:
assert 'time,cgm,bolus' in f.readline()
assert '2018-03-16 00:00:00,144.5,0.7,' in f.readline()
def test_file_exists(self):
with open('test_file1.csv', "w") as file_handle:
file_writer = csv.writer(file_handle,
delimiter=',',
quotechar='"',
quoting=csv.QUOTE_MINIMAL)
file_writer.writerow(['Id', 'time', 'value', 'no useful', 'info'])
Id = 1
check_vals = [[], [], []]
for i in range(100):
time = random_date()
Id += i
value = rdm.randint(100, 1000)
check_vals[0].append(Id)
check_vals[1].append(time)
check_vals[2].append(value)
file_writer.writerow([Id, time, value])
file_handle.close()
with open('test_file2.csv', "w") as file_handle:
file_writer = csv.writer(file_handle,
delimiter=',',
quotechar='"',
quoting=csv.QUOTE_MINIMAL)
file_writer.writerow(['Id', 'time', 'value', 'no useful', 'info'])
Id = 1
check_vals = [[], [], []]
for i in range(100):
time = random_date()
Id += i
value = rdm.randint(100, 1000)
check_vals[0].append(Id)
check_vals[1].append(time)
check_vals[2].append(value)
file_writer.writerow([Id, time, value])
file_handle.close()
imported_data1 = d_i.ImportData('test_file1.csv')
imported_data2 = d_i.ImportData('test_file2.csv')
zipp = [
imported_data1.roundTimeArray(5),
imported_data2.roundTimeArray(5)
]
d_i.printArray(zipp, ['test_file1.csv', 'test_file2.csv'],
'output_file', 'test_file1.csv')
self.assertTrue(path.exists('output_file.csv'))
os.remove('test_file1.csv')
os.remove('test_file2.csv')
def test_print_array_round_to_15(self):
data = []
for f in os.listdir('smallData'):
data.append(di.ImportData('smallData/' + f))
rdata_15 = []
for i in data:
rdata_15.append(di.roundTimeArray(i, 15))
self.assertNotEqual(
di.printArray(rdata_15, os.listdir('smallData'), 'out_15',
'meal_small.csv'), -1)
self.assertTrue(os.path.exists('out_15.csv'))
os.remove('out_15.csv')
def test_print_array(self):
files_lst = os.listdir('./smallData/')
data = []
for f in files_lst:
data.append(data_import.ImportData(join('./smallData/', f)))
data_5 = []
for obj in data:
data_5.append(data_import.roundTimeArray(obj, 5))
r = data_import.printArray(data_5, files_lst, 'out_5',
'smbg_small.csv')
self.assertTrue(os.path.exists('out_5.csv'))
os.remove('out_5.csv')
def test_print_array_round_to_5(self):
data = []
for f in os.listdir('smallData'):
if '.csv' in f:
data.append(data_import.ImportData('smallData/' + f))
data_5 = []
for i in data:
data_5.append(data_import.roundTimeArray(i, 5))
self.assertNotEqual(
data_import.printArray(data_5,
os.listdir('smallData'),
'out_5', 'cgm_small.csv'), -1)
self.assertTrue(os.path.exists('out_5.csv'))
os.remove('out_5.csv')
def test_print_array(self):
files_lst = os.listdir('smallData')
data_lst = []
for f in files_lst:
data_lst.append(data_import.ImportData('smallData/' + f))
data_5 = []
for obj in data_lst:
data_5.append(data_import.roundTimeArray(obj, 5))
r = data_import.printArray(data_5, files_lst, 'out_5', 'hr_small.csv')
self.assertNotEqual(r, -1)
self.assertTrue(os.path.exists('out_5.csv'))
os.remove('out_5.csv')
def test_printarray(self):
"""
A testing function for printArray
"""
file_lst = os.listdir('./smallData')
data_list = []
for i in range(len(file_lst)):
obj = data_import.ImportData('./smallData/' + file_lst[i])
data_list.append(data_import.roundTimeArray(obj, 5))
result = data_import.printArray(data_list, file_lst, 'test_printarray',
'meal_small.csv')
self.assertNotEqual(result, -1)
self.assertTrue(os.path.exists('test_printarray.csv'))
os.remove('test_printarray.csv')
# Linear Search (linear search is default)
t1 = time.time()
for data_import_obj in data_lst:
sum_key = [
add_file in data_import_obj._file_name
for add_file in ['activity, bolus, meal']
]
if any(sum_key):
data_5.append(
data_import.roundTimeArray(data_import_obj, 5,
operation='add'))
else:
data_5.append(data_import.roundTimeArray(data_import_obj, 5))
try:
data_import.printArray(data_5, files_lst, args.output_file + '_5.csv',
args.sort_key)
except IndexError:
print("sort_key provided was did not apply to the files in " +
args.folder_name,
file=sys.stderr)
sys.exit(1)
t2 = time.time()
for data_import_obj in data_lst:
sum_key = [
add_file in data_import_obj._file_name
for add_file in ['activity, bolus, meal']
]
if any(sum_key):
data_15.append(
