def test_get_run_data():
"""This just tests whether the .mat files return the same values as the .h5
files."""
runID = database.run_id_string(randint(0, 700))
pathToData = '/media/Data/Documents/School/UC Davis/Bicycle Mechanics/BicycleDAQ/data'
matFile = os.path.join(pathToData, runID + '.mat')
h5File = os.path.join(pathToData, 'h5', runID + '.h5')
matDat = database.get_run_data(matFile)
h5Dat = database.get_run_data(h5File)
for k, v in h5Dat.items():
if k == 'NICols' or k == 'VNavCols' or k == 'VNavDataText' :
assert v == matDat[k]
elif k == 'NIData' or k == 'VNavData':
npt.assert_allclose(v, matDat[k])
elif k == 'par':
for subKey, subVal in v.items():
if isinstance(subVal, type(ones(1))) or subKey == 'Speed':
npt.assert_allclose(subVal, matDat[k][subKey])
else:
assert subVal == matDat[k][subKey]
#print('{} matches'.format(subKey))
else:
assert v == matDat[k]
def test_get_calib_data():
"""This just tests whether the .mat files return the same values as the .h5
files."""
runID = database.run_id_string(randint(0, 20))
pathToData = '/media/Data/Documents/School/UC Davis/Bicycle Mechanics/BicycleDAQ/data/CalibData'
matFile = os.path.join(pathToData, runID + '.mat')
h5File = os.path.join(pathToData, 'h5', runID + '.h5')
matDat = database.get_calib_data(matFile)
h5Dat = database.get_calib_data(h5File)
for k, v in h5Dat.items():
if isinstance(v, type('')):
assert v == matDat[k]
else:
npt.assert_allclose(v, matDat[k])
def test_run_id_string():
ids = [105, '000105', '00105', '0105', '105']
for run in ids:
assert database.run_id_string(run) == '00105'