def test_loop_exp_frq_offset_point_time_cpmg_setup(self):
"""U{Bug #21665<https://gna.org/bugs/?21665>} catch, the failure due to a a CPMG analysis recorded at two fields at two delay times, using minimise.calculate()."""
# Load the state.
statefile = status.install_path + sep+'test_suite'+sep+'shared_data'+sep+'dispersion'+sep+'bug_21665.bz2'
state.load_state(statefile, force=True)
# Original data
ncyc_1 = [20, 16, 10, 36, 2, 12, 4, 22, 18, 40, 14, 26, 8, 32, 24, 6, 28]
sfrq_1 = 499.86214*1E6
time_T2_1 = 0.04
cpmg_1 = sorted([ncyc/time_T2_1 for ncyc in ncyc_1])
ncyc_2 = [28, 4, 32, 60, 2, 10, 16, 8, 20, 52, 18, 40, 6, 12, 24, 14, 22]
sfrq_2 = 599.8908587*1E6
time_T2_2 = 0.06
cpmg_2 = sorted([ncyc/time_T2_2 for ncyc in ncyc_2])
# Test the loop function.
# First initialize index for the two lists.
i = -1
j = -1
for exp_type, frq, offset, point, time, ei, mi, oi, di, ti in loop_exp_frq_offset_point_time(return_indices=True):
if frq == sfrq_1:
i += 1
self.assertEqual(time, time_T2_1)
self.assertAlmostEqual(point, cpmg_1[i], 3)
if frq == sfrq_2:
j += 1
self.assertEqual(time, time_T2_2)
self.assertAlmostEqual(point, cpmg_2[j], 3)
def test_loop_time_cpmg(self):
"""Unit test of the loop_time() function.
This uses the data of the saved state attached to U{bug #21665<https://gna.org/bugs/?21665>}.
"""
# Load the state.
statefile = status.install_path + sep+'test_suite'+sep+'shared_data'+sep+'dispersion'+sep+'bug_21665.bz2'
state.load_state(statefile, force=True)
# Original data (exp_type, frq, offset, point).
data = [
['SQ CPMG', 499862140.0, 0, [50.0, 100.0, 150.0, 200.0, 250.0, 300.0, 350.0, 400.0, 450.0, 500.0, 550.0, 600.0, 650.0, 700.0, 800.0, 900.0, 1000.0], 0.04],
['SQ CPMG', 599890858.69999993, 0, [33.3333, 66.666, 100.0, 133.333, 166.666, 200.0, 233.333, 266.666, 300.0, 333.333, 366.666, 400.0, 466.666, 533.333, 666.666, 866.666, 1000.0], 0.06]
]
# Original indices (ei, mi, oi, ti).
indices = [
[0, 0, 0, [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16], 0],
[0, 1, 0, [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16], 0]
]
# Check the number of iterations.
print("Checking the number of iterations of the loop.")
count_frq = 0
for dat in data:
frq = dat[1]
for time, ti in loop_time(frq=frq, return_indices=True):
print(time, ti)
count_frq += 1
self.assertEqual(count_frq, 2)
def test_loop_exp_frq_offset_point_cpmg(self):
"""Unit test of the loop_exp_frq_offset_point() function.
This uses the data of the saved state attached to U{bug #21665<https://gna.org/bugs/?21665>}.
"""
# Load the state.
statefile = status.install_path + sep+'test_suite'+sep+'shared_data'+sep+'dispersion'+sep+'bug_21665.bz2'
state.load_state(statefile, force=True)
# Original data (exp_type, frq, offset, point).
data = [
['SQ CPMG', 499862140.0, 0, [50.0, 100.0, 150.0, 200.0, 250.0, 300.0, 350.0, 400.0, 450.0, 500.0, 550.0, 600.0, 650.0, 700.0, 800.0, 900.0, 1000.0]],
['SQ CPMG', 599890858.69999993, 0, [33.3333, 66.666, 100.0, 133.333, 166.666, 200.0, 233.333, 266.666, 300.0, 333.333, 366.666, 400.0, 466.666, 533.333, 666.666, 866.666, 1000.0]]
]
# Original indices (ei, mi, oi).
indices = [
[0, 0, 0, [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16]],
[0, 1, 0, [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16]]
]
# Check the number of iterations.
print("Checking the number of iterations of the loop.")
count = 0
for exp_type, frq, offset, point, ei, mi, oi, di in loop_exp_frq_offset_point(return_indices=True):
print(exp_type, frq, offset, point, ei, mi, oi, di)
count += 1
self.assertEqual(count, 34)
# Check the values.
print("Checking the values returned by the loop.")
frq_index = 0
disp_index = 0
for exp_type, frq, offset, point, ei, mi, oi, di in loop_exp_frq_offset_point(return_indices=True):
# Check the experiment info.
self.assertEqual(exp_type, data[frq_index][0])
self.assertEqual(ei, indices[frq_index][0])
# Check the frequency info.
self.assertEqual(frq, data[frq_index][1])
self.assertEqual(mi, indices[frq_index][1])
# Check the offset info.
self.assertEqual(offset, data[frq_index][2])
self.assertEqual(oi, indices[frq_index][2])
# Check the dispersion point info.
self.assertAlmostEqual(point, data[frq_index][3][disp_index], 2)
self.assertEqual(di, indices[frq_index][3][disp_index])
# Increment the data index.
if disp_index == 16:
frq_index += 1
disp_index = 0
else:
disp_index += 1
def test_has_exponential_exp_type_cpmg(self):
"""Unit test of the has_exponential_exp_type() function.
This uses the data of the saved state attached to U{bug #21665<https://gna.org/bugs/?21665>}.
"""
# Load the state.
statefile = status.install_path + sep+'test_suite'+sep+'shared_data'+sep+'dispersion'+sep+'bug_21665.bz2'
state.load_state(statefile, force=True)
# Check the return of has_exponential_exp_type.
exponential_exp_type = has_exponential_exp_type()
print(exponential_exp_type)
self.assertEqual(exponential_exp_type, False)
def test_has_exponential_exp_type_cpmg(self):
"""Unit test of the has_exponential_exp_type() function.
This uses the data of the saved state attached to U{bug #21665<https://web.archive.org/web/https://gna.org/bugs/?21665>}.
"""
# Load the state.
statefile = status.install_path + sep + 'test_suite' + sep + 'shared_data' + sep + 'dispersion' + sep + 'bug_21665.bz2'
state.load_state(statefile, force=True)
# Check the return of has_exponential_exp_type.
exponential_exp_type = has_exponential_exp_type()
print(exponential_exp_type)
self.assertEqual(exponential_exp_type, False)
def test_get_curve_type_cpmg(self):
"""Unit test of the get_curve_type() function.
This uses the data of the saved state attached to U{bug #21665<https://web.archive.org/web/https://gna.org/bugs/?21665>}.
"""
# Load the state.
statefile = status.install_path + sep + 'test_suite' + sep + 'shared_data' + sep + 'dispersion' + sep + 'bug_21665.bz2'
state.load_state(statefile, force=True)
# Check the return of get_curve_type function.
curve_type = get_curve_type()
print(curve_type)
self.assertEqual(curve_type, 'fixed time')
def test_get_curve_type_cpmg(self):
"""Unit test of the get_curve_type() function.
This uses the data of the saved state attached to U{bug #21665<https://gna.org/bugs/?21665>}.
"""
# Load the state.
statefile = status.install_path + sep+'test_suite'+sep+'shared_data'+sep+'dispersion'+sep+'bug_21665.bz2'
state.load_state(statefile, force=True)
# Check the return of get_curve_type function.
curve_type = get_curve_type()
print(curve_type)
self.assertEqual(curve_type, 'fixed time')
def test_get_times_cpmg(self):
"""Unit test of the get_times() function.
This uses the data of the saved state attached to U{bug #21665<https://gna.org/bugs/?21665>}.
"""
# Load the state.
statefile = status.install_path + sep+'test_suite'+sep+'shared_data'+sep+'dispersion'+sep+'bug_21665.bz2'
state.load_state(statefile, force=True)
# Check the return of get_times().
times = get_times()
for exp_type in times:
print(times[exp_type])
self.assertEqual(len(times[exp_type]), 2)
def test_get_times_cpmg(self):
"""Unit test of the get_times() function.
This uses the data of the saved state attached to U{bug #21665<https://web.archive.org/web/https://gna.org/bugs/?21665>}.
"""
# Load the state.
statefile = status.install_path + sep + 'test_suite' + sep + 'shared_data' + sep + 'dispersion' + sep + 'bug_21665.bz2'
state.load_state(statefile, force=True)
# Check the return of get_times().
times = get_times()
for exp_type in times:
print(times[exp_type])
self.assertEqual(len(times[exp_type]), 2)
def test_loop_exp_frq_offset_cpmg(self):
"""Unit test of the loop_exp_frq_offset() function.
This uses the data of the saved state attached to U{bug #21665<https://gna.org/bugs/?21665>}.
"""
# Load the state.
statefile = status.install_path + sep+'test_suite'+sep+'shared_data'+sep+'dispersion'+sep+'bug_21665.bz2'
state.load_state(statefile, force=True)
# Original data (exp_type, frq, offset).
data = [
['SQ CPMG', 499862140.0, 0],
['SQ CPMG', 599890858.69999993, 0]
]
# Original indices (ei, mi, oi).
indices = [
[0, 0, 0],
[0, 1, 0]
]
# Check the number of iterations.
print("Checking the number of iterations of the loop.")
count = 0
for exp_type, frq, offset, ei, mi, oi in loop_exp_frq_offset(return_indices=True):
print(exp_type, frq, offset, ei, mi, oi)
count += 1
self.assertEqual(count, 2)
# Check the values.
print("Checking the values returned by the loop.")
index = 0
for exp_type, frq, offset, ei, mi, oi in loop_exp_frq_offset(return_indices=True):
# Check the experiment info.
self.assertEqual(exp_type, data[index][0])
self.assertEqual(ei, indices[index][0])
# Check the frequency info.
self.assertEqual(frq, data[index][1])
self.assertEqual(mi, indices[index][1])
# Check the offset info.
self.assertEqual(offset, data[index][2])
self.assertEqual(oi, indices[index][2])
# Increment the data index.
index += 1
def test_loop_time_cpmg(self):
"""Unit test of the loop_time() function.
This uses the data of the saved state attached to U{bug #21665<https://web.archive.org/web/https://gna.org/bugs/?21665>}.
"""
# Load the state.
statefile = status.install_path + sep + 'test_suite' + sep + 'shared_data' + sep + 'dispersion' + sep + 'bug_21665.bz2'
state.load_state(statefile, force=True)
# Original data (exp_type, frq, offset, point).
data = [[
'SQ CPMG', 499862140.0, 0,
[
50.0, 100.0, 150.0, 200.0, 250.0, 300.0, 350.0, 400.0, 450.0,
500.0, 550.0, 600.0, 650.0, 700.0, 800.0, 900.0, 1000.0
], 0.04
],
[
'SQ CPMG', 599890858.69999993, 0,
[
33.3333, 66.666, 100.0, 133.333, 166.666, 200.0,
233.333, 266.666, 300.0, 333.333, 366.666, 400.0,
466.666, 533.333, 666.666, 866.666, 1000.0
], 0.06
]]
# Original indices (ei, mi, oi, ti).
indices = [
[
0, 0, 0,
[0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16], 0
],
[
0, 1, 0,
[0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16], 0
]
]
# Check the number of iterations.
print("Checking the number of iterations of the loop.")
count_frq = 0
for dat in data:
frq = dat[1]
for time, ti in loop_time(frq=frq, return_indices=True):
print(time, ti)
count_frq += 1
self.assertEqual(count_frq, 2)
def test_loop_exp_frq_offset_cpmg(self):
"""Unit test of the loop_exp_frq_offset() function.
This uses the data of the saved state attached to U{bug #21665<https://web.archive.org/web/https://gna.org/bugs/?21665>}.
"""
# Load the state.
statefile = status.install_path + sep + 'test_suite' + sep + 'shared_data' + sep + 'dispersion' + sep + 'bug_21665.bz2'
state.load_state(statefile, force=True)
# Original data (exp_type, frq, offset).
data = [['SQ CPMG', 499862140.0, 0],
['SQ CPMG', 599890858.69999993, 0]]
# Original indices (ei, mi, oi).
indices = [[0, 0, 0], [0, 1, 0]]
# Check the number of iterations.
print("Checking the number of iterations of the loop.")
count = 0
for exp_type, frq, offset, ei, mi, oi in loop_exp_frq_offset(
return_indices=True):
print(exp_type, frq, offset, ei, mi, oi)
count += 1
self.assertEqual(count, 2)
# Check the values.
print("Checking the values returned by the loop.")
index = 0
for exp_type, frq, offset, ei, mi, oi in loop_exp_frq_offset(
return_indices=True):
# Check the experiment info.
self.assertEqual(exp_type, data[index][0])
self.assertEqual(ei, indices[index][0])
# Check the frequency info.
self.assertEqual(frq, data[index][1])
self.assertEqual(mi, indices[index][1])
# Check the offset info.
self.assertEqual(offset, data[index][2])
self.assertEqual(oi, indices[index][2])
# Increment the data index.
index += 1
def test_calc_rotating_frame_params(self):
"""Unit test of the calc_rotating_frame_params() function for R1rho setup.
This uses the data of the saved state attached to U{bug #21344<https://gna.org/bugs/?21344>}.
"""
# Load the state.
statefile = status.install_path + sep+'test_suite'+sep+'shared_data'+sep+'dispersion'+sep+'bug_21344_trunc.bz2'
state.load_state(statefile, force=True)
curspin_id = ':5@N'
curspin = return_spin(curspin_id)
# Use calc_tilt_angle function
theta_spin_dic, Domega_spin_dic, w_eff_spin_dic, dic_key_list = calc_rotating_frame_params(spin = curspin, spin_id = curspin_id)
for dic_key in dic_key_list:
print(dic_key, theta_spin_dic[dic_key], Domega_spin_dic[dic_key], w_eff_spin_dic[dic_key])
def test_loop_time_r1rho(self):
"""Unit test of the loop_time() function for R1rho setup.
This uses the data of the saved state attached to U{bug #21344<https://web.archive.org/web/https://gna.org/bugs/?21344>}.
"""
# Load the state.
statefile = status.install_path + sep + 'test_suite' + sep + 'shared_data' + sep + 'dispersion' + sep + 'bug_21344_trunc.bz2'
state.load_state(statefile, force=True)
# Offset and point combinations, with their associated time.
offset_point_time_list = [
[118.078, 431.0, [0.0, 0.04, 0.1, 0.2]],
[118.078, 651.2, [0.0, 0.04, 0.1, 0.2, 0.4]],
[118.078, 800.5, [0.0, 0.04, 0.1, 0.2, 0.4]],
[118.078, 984.0, [0.0, 0.04, 0.1, 0.2, 0.4]],
[118.078, 1341.11, [0.0, 0.04, 0.1, 0.2, 0.4]],
[118.078, 1648.5, [0.0, 0.04, 0.1, 0.14, 0.2]],
[124.247031462, 1341.11, [0.0, 0.04, 0.1, 0.2, 0.4]],
[130.416062924, 800.5, [0.0, 0.04, 0.1, 0.2, 0.4]],
[130.416062924, 1341.11, [0.0, 0.04, 0.1, 0.2, 0.4]],
[130.416062924, 1648.5, [0.0, 0.04, 0.1, 0.14, 0.2]],
[142.754125848, 800.5, [0.0, 0.04, 0.1, 0.2, 0.4]],
[142.754125848, 1341.11, [0.0, 0.04, 0.1, 0.2, 0.4]],
[179.768314621, 1341.11, [0.0, 0.04, 0.1, 0.2, 0.4]],
[241.458629241, 1341.11, [0.0, 0.04, 0.1, 0.2, 0.4]]
]
# Check the number of iterations.
print("Checking the number of iterations of the loop.")
frq = 799777399.1
for offset, point, time_list in offset_point_time_list:
# Define count
count = 0
for time, ti in loop_time(frq=frq,
offset=offset,
point=point,
return_indices=True):
print(frq, offset, point, time, ti, count)
self.assertEqual(time, time_list[count])
self.assertEqual(ti, count)
count += 1
def test_calc_rotating_frame_params(self):
"""Unit test of the calc_rotating_frame_params() function for R1rho setup.
This uses the data of the saved state attached to U{bug #21344<https://web.archive.org/web/https://gna.org/bugs/?21344>}.
"""
# Load the state.
statefile = status.install_path + sep + 'test_suite' + sep + 'shared_data' + sep + 'dispersion' + sep + 'bug_21344_trunc.bz2'
state.load_state(statefile, force=True)
curspin_id = ':5@N'
curspin = return_spin(spin_id=curspin_id)
# Use calc_tilt_angle function
theta_spin_dic, Domega_spin_dic, w_eff_spin_dic, dic_key_list = calc_rotating_frame_params(
spin=curspin, spin_id=curspin_id)
for dic_key in dic_key_list:
print(dic_key, theta_spin_dic[dic_key], Domega_spin_dic[dic_key],
w_eff_spin_dic[dic_key])
def test_loop_time_r1rho(self):
"""Unit test of the loop_time() function for R1rho setup.
This uses the data of the saved state attached to U{bug #21344<https://gna.org/bugs/?21344>}.
"""
# Load the state.
statefile = status.install_path + sep+'test_suite'+sep+'shared_data'+sep+'dispersion'+sep+'bug_21344_trunc.bz2'
state.load_state(statefile, force=True)
# Offset and point combinations, with their associated time.
offset_point_time_list = [
[118.078, 431.0, [0.0, 0.04, 0.1, 0.2]],
[118.078, 651.2, [0.0, 0.04, 0.1, 0.2, 0.4]],
[118.078, 800.5, [0.0, 0.04, 0.1, 0.2, 0.4]],
[118.078, 984.0, [0.0, 0.04, 0.1, 0.2, 0.4]],
[118.078, 1341.11, [0.0, 0.04, 0.1, 0.2, 0.4]],
[118.078, 1648.5, [0.0, 0.04, 0.1, 0.14, 0.2]],
[124.247031462, 1341.11, [0.0, 0.04, 0.1, 0.2, 0.4]],
[130.416062924, 800.5, [0.0, 0.04, 0.1, 0.2, 0.4]],
[130.416062924, 1341.11, [0.0, 0.04, 0.1, 0.2, 0.4]],
[130.416062924, 1648.5, [0.0, 0.04, 0.1, 0.14, 0.2]],
[142.754125848, 800.5, [0.0, 0.04, 0.1, 0.2, 0.4]],
[142.754125848, 1341.11, [0.0, 0.04, 0.1, 0.2, 0.4]],
[179.768314621, 1341.11, [0.0, 0.04, 0.1, 0.2, 0.4]],
[241.458629241, 1341.11, [0.0, 0.04, 0.1, 0.2, 0.4]]]
# Check the number of iterations.
print("Checking the number of iterations of the loop.")
frq = 799777399.1
for offset, point, time_list in offset_point_time_list:
# Define count
count = 0
for time, ti in loop_time(frq=frq, offset=offset, point=point, return_indices=True):
print(frq, offset, point, time, ti, count)
self.assertEqual(time, time_list[count])
self.assertEqual(ti, count)
count += 1
def test_count_relax_times_cpmg(self):
"""Unit test of the count_relax_times() function.
This uses the data of the saved state attached to U{bug #21665<https://gna.org/bugs/?21665>}.
"""
# Load the state.
statefile = status.install_path + sep+'test_suite'+sep+'shared_data'+sep+'dispersion'+sep+'bug_21665.bz2'
state.load_state(statefile, force=True)
# Original data (exp_type, frq).
data = [
['SQ CPMG', 499862140.0],
['SQ CPMG', 599890858.69999993]
]
# Original indices (ei, mi).
indices = [
[0, 0],
[0, 1]
]
# Check the number of time counts.
print("Checking the number of time counts.")
for id in cdp.exp_type:
exp_type = cdp.exp_type[id]
frq = cdp.spectrometer_frq[id]
point = cdp.cpmg_frqs[id]
count = count_relax_times(exp_type = exp_type, frq = frq, point = point, ei = cdp.exp_type_list.index(cdp.exp_type[id]))
print(id, exp_type, frq, point, count)
# Test the data
if id.split('A')[0] == 'Z_':
self.assertEqual(exp_type, data[0][0])
self.assertEqual(frq, data[0][1])
elif id.split('B')[0] == 'Z_':
self.assertEqual(exp_type, data[1][0])
self.assertEqual(frq, data[1][1])
# Test the time count
self.assertEqual(count, 1)
def test_count_relax_times_cpmg(self):
"""Unit test of the count_relax_times() function.
This uses the data of the saved state attached to U{bug #21665<https://web.archive.org/web/https://gna.org/bugs/?21665>}.
"""
# Load the state.
statefile = status.install_path + sep + 'test_suite' + sep + 'shared_data' + sep + 'dispersion' + sep + 'bug_21665.bz2'
state.load_state(statefile, force=True)
# Original data (exp_type, frq).
data = [['SQ CPMG', 499862140.0], ['SQ CPMG', 599890858.69999993]]
# Original indices (ei, mi).
indices = [[0, 0], [0, 1]]
# Check the number of time counts.
print("Checking the number of time counts.")
for id in cdp.exp_type:
exp_type = cdp.exp_type[id]
frq = cdp.spectrometer_frq[id]
point = cdp.cpmg_frqs[id]
count = count_relax_times(exp_type=exp_type,
frq=frq,
point=point,
ei=cdp.exp_type_list.index(
cdp.exp_type[id]))
print(id, exp_type, frq, point, count)
# Test the data
if id.split('A')[0] == 'Z_':
self.assertEqual(exp_type, data[0][0])
self.assertEqual(frq, data[0][1])
elif id.split('B')[0] == 'Z_':
self.assertEqual(exp_type, data[1][0])
self.assertEqual(frq, data[1][1])
# Test the time count
self.assertEqual(count, 1)
def test_loop_exp_frq_offset_point_time_cpmg_setup(self):
"""U{Bug #21665<https://web.archive.org/web/https://gna.org/bugs/?21665>} catch, the failure due to a a CPMG analysis recorded at two fields at two delay times, using minimise.calculate()."""
# Load the state.
statefile = status.install_path + sep + 'test_suite' + sep + 'shared_data' + sep + 'dispersion' + sep + 'bug_21665.bz2'
state.load_state(statefile, force=True)
# Original data
ncyc_1 = [
20, 16, 10, 36, 2, 12, 4, 22, 18, 40, 14, 26, 8, 32, 24, 6, 28
]
sfrq_1 = 499.86214 * 1E6
time_T2_1 = 0.04
cpmg_1 = sorted([ncyc / time_T2_1 for ncyc in ncyc_1])
ncyc_2 = [
28, 4, 32, 60, 2, 10, 16, 8, 20, 52, 18, 40, 6, 12, 24, 14, 22
]
sfrq_2 = 599.8908587 * 1E6
time_T2_2 = 0.06
cpmg_2 = sorted([ncyc / time_T2_2 for ncyc in ncyc_2])
# Test the loop function.
# First initialize index for the two lists.
i = -1
j = -1
for exp_type, frq, offset, point, time, ei, mi, oi, di, ti in loop_exp_frq_offset_point_time(
return_indices=True):
if frq == sfrq_1:
i += 1
self.assertEqual(time, time_T2_1)
self.assertAlmostEqual(point, cpmg_1[i], 3)
if frq == sfrq_2:
j += 1
self.assertEqual(time, time_T2_2)
self.assertAlmostEqual(point, cpmg_2[j], 3)
def test_return_offset_data(self):
"""Unit test of the return_offset_data() function for R1rho setup.
This uses the data of the saved state attached to U{bug #21344<https://web.archive.org/web/https://gna.org/bugs/?21344>}.
"""
# Load the state.
statefile = status.install_path + sep + 'test_suite' + sep + 'shared_data' + sep + 'dispersion' + sep + 'bug_21344_trunc.bz2'
state.load_state(statefile, force=True)
# Get the field count
field_count = cdp.spectrometer_frq_count
# Get the spin_lock_field points
spin_lock_nu1 = return_spin_lock_nu1(ref_flag=False)
# Initialize data containers
all_spin_ids = get_spin_ids()
# Containers for only selected spins
cur_spin_ids = []
cur_spins = []
for curspin_id in all_spin_ids:
# Get the spin
curspin = return_spin(spin_id=curspin_id)
# Check if is selected
if curspin.select == True:
cur_spin_ids.append(curspin_id)
cur_spins.append(curspin)
# Get the spectometer frequency
sfrq = cdp.spectrometer_frq_list[0]
# Gyromagnetic ratios
g1H = 26.7522212 * 1e7
g15N = -2.7126 * 1e7
# The offset and R1 data.
offsets, spin_lock_fields_inter, chemical_shifts, tilt_angles, Delta_omega, w_eff = return_offset_data(
spins=cur_spins,
spin_ids=cur_spin_ids,
field_count=field_count,
fields=spin_lock_nu1)
# Loop over the index of spins, then exp_type, frq, offset
print("Printing the following")
print(
"exp_type curspin_id frq offset{ppm} offsets[ei][si][mi][oi]{rad/s} ei mi oi si di cur_spin.chemical_shift{ppm} chemical_shifts[ei][si][mi]{rad/s} spin_lock_nu1{Hz} tilt_angles[ei][si][mi][oi]{rad}"
)
for si in range(len(cur_spin_ids)):
curspin_id = cur_spin_ids[si]
cur_spin = cur_spins[si]
for exp_type, frq, offset, ei, mi, oi in loop_exp_frq_offset(
return_indices=True):
# Loop over the dispersion points.
spin_lock_fields = spin_lock_nu1[ei][mi][oi]
for di in range(len(spin_lock_fields)):
print(
"%-8s %-10s %11.1f %8.4f %12.5f %i %i %i %i %i %7.3f %12.5f %12.5f %12.5f"
% (exp_type, curspin_id, frq, offset,
offsets[ei][si][mi][oi], ei, mi, oi, si, di,
cur_spin.chemical_shift,
chemical_shifts[ei][si][mi], spin_lock_fields[di],
tilt_angles[ei][si][mi][oi][di]))
# Test chemical shift conversion
self.assertEqual(
chemical_shifts[ei][si][mi], cur_spin.chemical_shift *
2.0 * pi * sfrq / g1H * g15N * 1e-6)
# Test the offset conversion
self.assertEqual(
offsets[ei][si][mi][oi],
offset * 2.0 * pi * sfrq / g1H * g15N * 1e-6)
# Test the tiltangle
c_omega1 = spin_lock_fields[di] * 2.0 * pi
c_Delta_omega = chemical_shifts[ei][si][mi] - offsets[ei][
si][mi][oi]
if c_omega1 / c_Delta_omega > 0:
c_theta = atan(c_omega1 / c_Delta_omega)
else:
c_theta = pi + atan(c_omega1 / c_Delta_omega)
self.assertAlmostEqual(tilt_angles[ei][si][mi][oi][di],
c_theta, 15)
def test_return_offset_data(self):
"""Unit test of the return_offset_data() function for R1rho setup.
This uses the data of the saved state attached to U{bug #21344<https://gna.org/bugs/?21344>}.
"""
# Load the state.
statefile = status.install_path + sep+'test_suite'+sep+'shared_data'+sep+'dispersion'+sep+'bug_21344_trunc.bz2'
state.load_state(statefile, force=True)
# Get the field count
field_count = cdp.spectrometer_frq_count
# Get the spin_lock_field points
spin_lock_nu1 = return_spin_lock_nu1(ref_flag=False)
# Initialize data containers
all_spin_ids = get_spin_ids()
# Containers for only selected spins
cur_spin_ids = []
cur_spins = []
for curspin_id in all_spin_ids:
# Get the spin
curspin = return_spin(curspin_id)
# Check if is selected
if curspin.select == True:
cur_spin_ids.append(curspin_id)
cur_spins.append(curspin)
# Get the spectometer frequency
sfrq = cdp.spectrometer_frq_list[0]
# Gyromagnetic ratios
g1H = 26.7522212 * 1e7
g15N = -2.7126 * 1e7
# The offset and R1 data.
offsets, spin_lock_fields_inter, chemical_shifts, tilt_angles, Delta_omega, w_eff = return_offset_data(spins=cur_spins, spin_ids=cur_spin_ids, field_count=field_count, fields=spin_lock_nu1)
# Loop over the index of spins, then exp_type, frq, offset
print("Printing the following")
print("exp_type curspin_id frq offset{ppm} offsets[ei][si][mi][oi]{rad/s} ei mi oi si di cur_spin.chemical_shift{ppm} chemical_shifts[ei][si][mi]{rad/s} spin_lock_nu1{Hz} tilt_angles[ei][si][mi][oi]{rad}")
for si in range(len(cur_spin_ids)):
curspin_id = cur_spin_ids[si]
cur_spin = cur_spins[si]
for exp_type, frq, offset, ei, mi, oi in loop_exp_frq_offset(return_indices=True):
# Loop over the dispersion points.
spin_lock_fields = spin_lock_nu1[ei][mi][oi]
for di in range(len(spin_lock_fields)):
print("%-8s %-10s %11.1f %8.4f %12.5f %i %i %i %i %i %7.3f %12.5f %12.5f %12.5f"%(exp_type, curspin_id, frq, offset, offsets[ei][si][mi][oi], ei, mi, oi, si, di, cur_spin.chemical_shift, chemical_shifts[ei][si][mi], spin_lock_fields[di], tilt_angles[ei][si][mi][oi][di]))
# Test chemical shift conversion
self.assertEqual(chemical_shifts[ei][si][mi], cur_spin.chemical_shift * 2.0 * pi * sfrq / g1H * g15N * 1e-6)
# Test the offset conversion
self.assertEqual(offsets[ei][si][mi][oi], offset * 2.0 * pi * sfrq / g1H * g15N * 1e-6)
# Test the tiltangle
c_omega1 = spin_lock_fields[di] * 2.0 * pi
c_Delta_omega = chemical_shifts[ei][si][mi] - offsets[ei][si][mi][oi]
if c_omega1 / c_Delta_omega > 0:
c_theta = atan(c_omega1 / c_Delta_omega)
else:
c_theta = pi + atan(c_omega1 / c_Delta_omega)
self.assertAlmostEqual(tilt_angles[ei][si][mi][oi][di], c_theta, 15)
def test_count_relax_times_r1rho(self):
"""Unit test of the count_relax_times() function.
This uses the data of the saved state attached to U{bug #21344<https://web.archive.org/web/https://gna.org/bugs/?21344>}.
"""
# Load the state.
statefile = status.install_path + sep + 'test_suite' + sep + 'shared_data' + sep + 'dispersion' + sep + 'bug_21344_trunc.bz2'
state.load_state(statefile, force=True)
# Original data (spectrum id: exp_type, frq, omega_rf_ppm, spin_lock_field_strength, time_spin_lock).
data = dict()
data['46_0_35_0'] = ['R1rho', 799777399.1, 118.078, 431.0, 0.0]
data['48_0_35_4'] = ['R1rho', 799777399.1, 118.078, 431.0, 0.04]
data['47_0_35_10'] = ['R1rho', 799777399.1, 118.078, 431.0, 0.1]
data['49_0_35_20'] = ['R1rho', 799777399.1, 118.078, 431.0, 0.2]
data['36_0_39_0'] = ['R1rho', 799777399.1, 118.078, 651.2, 0.0]
data['39_0_39_4'] = ['R1rho', 799777399.1, 118.078, 651.2, 0.04]
data['37_0_39_10'] = ['R1rho', 799777399.1, 118.078, 651.2, 0.1]
data['40_0_39_20'] = ['R1rho', 799777399.1, 118.078, 651.2, 0.2]
data['38_0_39_40'] = ['R1rho', 799777399.1, 118.078, 651.2, 0.4]
data['41_0_41_0'] = ['R1rho', 799777399.1, 118.078, 800.5, 0.0]
data['44_0_41_4'] = ['R1rho', 799777399.1, 118.078, 800.5, 0.04]
data['42_0_41_10'] = ['R1rho', 799777399.1, 118.078, 800.5, 0.1]
data['45_0_41_20'] = ['R1rho', 799777399.1, 118.078, 800.5, 0.2]
data['43_0_41_40'] = ['R1rho', 799777399.1, 118.078, 800.5, 0.4]
data['31_0_43_0'] = ['R1rho', 799777399.1, 118.078, 984.0, 0.0]
data['34_0_43_4'] = ['R1rho', 799777399.1, 118.078, 984.0, 0.04]
data['32_0_43_10'] = ['R1rho', 799777399.1, 118.078, 984.0, 0.1]
data['35_0_43_20'] = ['R1rho', 799777399.1, 118.078, 984.0, 0.2]
data['33_0_43_40'] = ['R1rho', 799777399.1, 118.078, 984.0, 0.4]
data['1_0_46_0'] = ['R1rho', 799777399.1, 118.078, 1341.11, 0.0]
data['4_0_46_4'] = ['R1rho', 799777399.1, 118.078, 1341.11, 0.04]
data['2_0_46_10'] = ['R1rho', 799777399.1, 118.078, 1341.11, 0.1]
data['5_0_46_20'] = ['R1rho', 799777399.1, 118.078, 1341.11, 0.2]
data['3_0_46_40'] = ['R1rho', 799777399.1, 118.078, 1341.11, 0.4]
data['60_0_48_0'] = ['R1rho', 799777399.1, 118.078, 1648.5, 0.0]
data['63_0_48_4'] = ['R1rho', 799777399.1, 118.078, 1648.5, 0.04]
data['61_0_48_10'] = ['R1rho', 799777399.1, 118.078, 1648.5, 0.1]
data['62_0_48_14'] = ['R1rho', 799777399.1, 118.078, 1648.5, 0.14]
data['64_0_48_20'] = ['R1rho', 799777399.1, 118.078, 1648.5, 0.2]
data['11_500_46_0'] = [
'R1rho', 799777399.1, 124.24703146206046, 1341.11, 0.0
]
data['14_500_46_4'] = [
'R1rho', 799777399.1, 124.24703146206046, 1341.11, 0.04
]
data['12_500_46_10'] = [
'R1rho', 799777399.1, 124.24703146206046, 1341.11, 0.1
]
data['15_500_46_20'] = [
'R1rho', 799777399.1, 124.24703146206046, 1341.11, 0.2
]
data['13_500_46_40'] = [
'R1rho', 799777399.1, 124.24703146206046, 1341.11, 0.4
]
data['50_1000_41_0'] = [
'R1rho', 799777399.1, 130.41606292412092, 800.5, 0.0
]
data['53_1000_41_4'] = [
'R1rho', 799777399.1, 130.41606292412092, 800.5, 0.04
]
data['51_1000_41_10'] = [
'R1rho', 799777399.1, 130.41606292412092, 800.5, 0.1
]
data['54_1000_41_20'] = [
'R1rho', 799777399.1, 130.41606292412092, 800.5, 0.2
]
data['52_1000_41_40'] = [
'R1rho', 799777399.1, 130.41606292412092, 800.5, 0.4
]
data['21_1000_46_0'] = [
'R1rho', 799777399.1, 130.41606292412092, 1341.11, 0.0
]
data['24_1000_46_4'] = [
'R1rho', 799777399.1, 130.41606292412092, 1341.11, 0.04
]
data['22_1000_46_10'] = [
'R1rho', 799777399.1, 130.41606292412092, 1341.11, 0.1
]
data['25_1000_46_20'] = [
'R1rho', 799777399.1, 130.41606292412092, 1341.11, 0.2
]
data['23_1000_46_40'] = [
'R1rho', 799777399.1, 130.41606292412092, 1341.11, 0.4
]
data['65_1000_48_0'] = [
'R1rho', 799777399.1, 130.41606292412092, 1648.5, 0.0
]
data['68_1000_48_4'] = [
'R1rho', 799777399.1, 130.41606292412092, 1648.5, 0.04
]
data['66_1000_48_10'] = [
'R1rho', 799777399.1, 130.41606292412092, 1648.5, 0.1
]
data['67_1000_48_14'] = [
'R1rho', 799777399.1, 130.41606292412092, 1648.5, 0.14
]
data['69_1000_48_20'] = [
'R1rho', 799777399.1, 130.41606292412092, 1648.5, 0.2
]
data['55_2000_41_0'] = [
'R1rho', 799777399.1, 142.75412584824184, 800.5, 0.0
]
data['58_2000_41_4'] = [
'R1rho', 799777399.1, 142.75412584824184, 800.5, 0.04
]
data['56_2000_41_10'] = [
'R1rho', 799777399.1, 142.75412584824184, 800.5, 0.1
]
data['59_2000_41_20'] = [
'R1rho', 799777399.1, 142.75412584824184, 800.5, 0.2
]
data['57_2000_41_40'] = [
'R1rho', 799777399.1, 142.75412584824184, 800.5, 0.4
]
data['6_2000_46_0'] = [
'R1rho', 799777399.1, 142.75412584824184, 1341.11, 0.0
]
data['9_2000_46_4'] = [
'R1rho', 799777399.1, 142.75412584824184, 1341.11, 0.04
]
data['7_2000_46_10'] = [
'R1rho', 799777399.1, 142.75412584824184, 1341.11, 0.1
]
data['10_2000_46_20'] = [
'R1rho', 799777399.1, 142.75412584824184, 1341.11, 0.2
]
data['8_2000_46_40'] = [
'R1rho', 799777399.1, 142.75412584824184, 1341.11, 0.4
]
data['16_5000_46_0'] = [
'R1rho', 799777399.1, 179.76831462060457, 1341.11, 0.0
]
data['19_5000_46_4'] = [
'R1rho', 799777399.1, 179.76831462060457, 1341.11, 0.04
]
data['17_5000_46_10'] = [
'R1rho', 799777399.1, 179.76831462060457, 1341.11, 0.1
]
data['20_5000_46_20'] = [
'R1rho', 799777399.1, 179.76831462060457, 1341.11, 0.2
]
data['18_5000_46_40'] = [
'R1rho', 799777399.1, 179.76831462060457, 1341.11, 0.4
]
data['26_10000_46_0'] = [
'R1rho', 799777399.1, 241.45862924120914, 1341.11, 0.0
]
data['29_10000_46_4'] = [
'R1rho', 799777399.1, 241.45862924120914, 1341.11, 0.04
]
data['27_10000_46_10'] = [
'R1rho', 799777399.1, 241.45862924120914, 1341.11, 0.1
]
data['30_10000_46_20'] = [
'R1rho', 799777399.1, 241.45862924120914, 1341.11, 0.2
]
data['28_10000_46_40'] = [
'R1rho', 799777399.1, 241.45862924120914, 1341.11, 0.4
]
time_comp = {
'118.078_431.00': 4,
'118.078_651.20': 5,
'118.078_800.50': 5,
'118.078_984.00': 5,
'118.078_1341.11': 5,
'118.078_1648.50': 5,
'124.247_1341.11': 5,
'130.416_800.50': 5,
'130.416_1341.11': 5,
'130.416_1648.50': 5,
'142.754_800.50': 5,
'142.754_1341.11': 5,
'179.768_1341.11': 5,
'241.459_1341.11': 5
}
# Check the number of time counts.
print("Checking the number of time counts.")
for id in cdp.exp_type:
exp_type = cdp.exp_type[id]
frq = cdp.spectrometer_frq[id]
offset = cdp.spin_lock_offset[id]
point = cdp.spin_lock_nu1[id]
count = count_relax_times(exp_type=exp_type,
frq=frq,
offset=offset,
point=point,
ei=cdp.exp_type_list.index(
cdp.exp_type[id]))
print(id, exp_type, frq, offset, point, count)
# Test the time count
print(time_comp)
self.assertEqual(count, time_comp['%.3f_%.2f' % (offset, point)])
def test_find_intensity_keys_r1rho(self):
"""Unit test of the find_intensity_keys() function.
This uses the data of the saved state attached to U{bug #21344<https://web.archive.org/web/https://gna.org/bugs/?21344>}.
"""
# Load the state.
statefile = status.install_path + sep + 'test_suite' + sep + 'shared_data' + sep + 'dispersion' + sep + 'bug_21344_trunc.bz2'
state.load_state(statefile, force=True)
# Original data (spectrum id: exp_type, frq, omega_rf_ppm, spin_lock_field_strength, time_spin_lock).
data = dict()
data['118_431.00_0.00'] = [
'46_0_35_0', 'R1rho', 799777399.1, 118.078, 431.0, 0.0
]
data['118_431.00_0.04'] = [
'48_0_35_4', 'R1rho', 799777399.1, 118.078, 431.0, 0.04
]
data['118_431.00_0.10'] = [
'47_0_35_10', 'R1rho', 799777399.1, 118.078, 431.0, 0.1
]
data['118_431.00_0.20'] = [
'49_0_35_20', 'R1rho', 799777399.1, 118.078, 431.0, 0.2
]
data['118_651.20_0.00'] = [
'36_0_39_0', 'R1rho', 799777399.1, 118.078, 651.2, 0.0
]
data['118_651.20_0.04'] = [
'39_0_39_4', 'R1rho', 799777399.1, 118.078, 651.2, 0.04
]
data['118_651.20_0.10'] = [
'37_0_39_10', 'R1rho', 799777399.1, 118.078, 651.2, 0.1
]
data['118_651.20_0.20'] = [
'40_0_39_20', 'R1rho', 799777399.1, 118.078, 651.2, 0.2
]
data['118_651.20_0.40'] = [
'38_0_39_40', 'R1rho', 799777399.1, 118.078, 651.2, 0.4
]
data['118_800.50_0.00'] = [
'41_0_41_0', 'R1rho', 799777399.1, 118.078, 800.5, 0.0
]
data['118_800.50_0.04'] = [
'44_0_41_4', 'R1rho', 799777399.1, 118.078, 800.5, 0.04
]
data['118_800.50_0.10'] = [
'42_0_41_10', 'R1rho', 799777399.1, 118.078, 800.5, 0.1
]
data['118_800.50_0.20'] = [
'45_0_41_20', 'R1rho', 799777399.1, 118.078, 800.5, 0.2
]
data['118_800.50_0.40'] = [
'43_0_41_40', 'R1rho', 799777399.1, 118.078, 800.5, 0.4
]
data['118_984.00_0.00'] = [
'31_0_43_0', 'R1rho', 799777399.1, 118.078, 984.0, 0.0
]
data['118_984.00_0.04'] = [
'34_0_43_4', 'R1rho', 799777399.1, 118.078, 984.0, 0.04
]
data['118_984.00_0.10'] = [
'32_0_43_10', 'R1rho', 799777399.1, 118.078, 984.0, 0.1
]
data['118_984.00_0.20'] = [
'35_0_43_20', 'R1rho', 799777399.1, 118.078, 984.0, 0.2
]
data['118_984.00_0.40'] = [
'33_0_43_40', 'R1rho', 799777399.1, 118.078, 984.0, 0.4
]
data['118_1341.11_0.00'] = [
'1_0_46_0', 'R1rho', 799777399.1, 118.078, 1341.11, 0.0
]
data['118_1341.11_0.04'] = [
'4_0_46_4', 'R1rho', 799777399.1, 118.078, 1341.11, 0.04
]
data['118_1341.11_0.10'] = [
'2_0_46_10', 'R1rho', 799777399.1, 118.078, 1341.11, 0.1
]
data['118_1341.11_0.20'] = [
'5_0_46_20', 'R1rho', 799777399.1, 118.078, 1341.11, 0.2
]
data['118_1341.11_0.40'] = [
'3_0_46_40', 'R1rho', 799777399.1, 118.078, 1341.11, 0.4
]
data['118_1648.50_0.00'] = [
'60_0_48_0', 'R1rho', 799777399.1, 118.078, 1648.5, 0.0
]
data['118_1648.50_0.04'] = [
'63_0_48_4', 'R1rho', 799777399.1, 118.078, 1648.5, 0.04
]
data['118_1648.50_0.10'] = [
'61_0_48_10', 'R1rho', 799777399.1, 118.078, 1648.5, 0.1
]
data['118_1648.50_0.14'] = [
'62_0_48_14', 'R1rho', 799777399.1, 118.078, 1648.5, 0.14
]
data['118_1648.50_0.20'] = [
'64_0_48_20', 'R1rho', 799777399.1, 118.078, 1648.5, 0.2
]
data['124_1341.11_0.00'] = [
'11_500_46_0', 'R1rho', 799777399.1, 124.247031462, 1341.11, 0.0
]
data['124_1341.11_0.04'] = [
'14_500_46_4', 'R1rho', 799777399.1, 124.247031462, 1341.11, 0.04
]
data['124_1341.11_0.10'] = [
'12_500_46_10', 'R1rho', 799777399.1, 124.247031462, 1341.11, 0.1
]
data['124_1341.11_0.20'] = [
'15_500_46_20', 'R1rho', 799777399.1, 124.247031462, 1341.11, 0.2
]
data['124_1341.11_0.40'] = [
'13_500_46_40', 'R1rho', 799777399.1, 124.247031462, 1341.11, 0.4
]
data['130_800.50_0.00'] = [
'50_1000_41_0', 'R1rho', 799777399.1, 130.416062924, 800.5, 0.0
]
data['130_800.50_0.04'] = [
'53_1000_41_4', 'R1rho', 799777399.1, 130.416062924, 800.5, 0.04
]
data['130_800.50_0.10'] = [
'51_1000_41_10', 'R1rho', 799777399.1, 130.416062924, 800.5, 0.1
]
data['130_800.50_0.20'] = [
'54_1000_41_20', 'R1rho', 799777399.1, 130.416062924, 800.5, 0.2
]
data['130_800.50_0.40'] = [
'52_1000_41_40', 'R1rho', 799777399.1, 130.416062924, 800.5, 0.4
]
data['130_1341.11_0.00'] = [
'21_1000_46_0', 'R1rho', 799777399.1, 130.416062924, 1341.11, 0.0
]
data['130_1341.11_0.04'] = [
'24_1000_46_4', 'R1rho', 799777399.1, 130.416062924, 1341.11, 0.04
]
data['130_1341.11_0.10'] = [
'22_1000_46_10', 'R1rho', 799777399.1, 130.416062924, 1341.11, 0.1
]
data['130_1341.11_0.20'] = [
'25_1000_46_20', 'R1rho', 799777399.1, 130.416062924, 1341.11, 0.2
]
data['130_1341.11_0.40'] = [
'23_1000_46_40', 'R1rho', 799777399.1, 130.416062924, 1341.11, 0.4
]
data['130_1648.50_0.00'] = [
'65_1000_48_0', 'R1rho', 799777399.1, 130.416062924, 1648.5, 0.0
]
data['130_1648.50_0.04'] = [
'68_1000_48_4', 'R1rho', 799777399.1, 130.416062924, 1648.5, 0.04
]
data['130_1648.50_0.10'] = [
'66_1000_48_10', 'R1rho', 799777399.1, 130.416062924, 1648.5, 0.1
]
data['130_1648.50_0.14'] = [
'67_1000_48_14', 'R1rho', 799777399.1, 130.416062924, 1648.5, 0.14
]
data['130_1648.50_0.20'] = [
'69_1000_48_20', 'R1rho', 799777399.1, 130.416062924, 1648.5, 0.2
]
data['143_800.50_0.00'] = [
'55_2000_41_0', 'R1rho', 799777399.1, 142.754125848, 800.5, 0.0
]
data['143_800.50_0.04'] = [
'58_2000_41_4', 'R1rho', 799777399.1, 142.754125848, 800.5, 0.04
]
data['143_800.50_0.10'] = [
'56_2000_41_10', 'R1rho', 799777399.1, 142.754125848, 800.5, 0.1
]
data['143_800.50_0.20'] = [
'59_2000_41_20', 'R1rho', 799777399.1, 142.754125848, 800.5, 0.2
]
data['143_800.50_0.40'] = [
'57_2000_41_40', 'R1rho', 799777399.1, 142.754125848, 800.5, 0.4
]
data['143_1341.11_0.00'] = [
'6_2000_46_0', 'R1rho', 799777399.1, 142.754125848, 1341.11, 0.0
]
data['143_1341.11_0.04'] = [
'9_2000_46_4', 'R1rho', 799777399.1, 142.754125848, 1341.11, 0.04
]
data['143_1341.11_0.10'] = [
'7_2000_46_10', 'R1rho', 799777399.1, 142.754125848, 1341.11, 0.1
]
data['143_1341.11_0.20'] = [
'10_2000_46_20', 'R1rho', 799777399.1, 142.754125848, 1341.11, 0.2
]
data['143_1341.11_0.40'] = [
'8_2000_46_40', 'R1rho', 799777399.1, 142.754125848, 1341.11, 0.4
]
data['180_1341.11_0.00'] = [
'16_5000_46_0', 'R1rho', 799777399.1, 179.768314621, 1341.11, 0.0
]
data['180_1341.11_0.04'] = [
'19_5000_46_4', 'R1rho', 799777399.1, 179.768314621, 1341.11, 0.04
]
data['180_1341.11_0.10'] = [
'17_5000_46_10', 'R1rho', 799777399.1, 179.768314621, 1341.11, 0.1
]
data['180_1341.11_0.20'] = [
'20_5000_46_20', 'R1rho', 799777399.1, 179.768314621, 1341.11, 0.2
]
data['180_1341.11_0.40'] = [
'18_5000_46_40', 'R1rho', 799777399.1, 179.768314621, 1341.11, 0.4
]
data['241_1341.11_0.00'] = [
'26_10000_46_0', 'R1rho', 799777399.1, 241.458629241, 1341.11, 0.0
]
data['241_1341.11_0.04'] = [
'29_10000_46_4', 'R1rho', 799777399.1, 241.458629241, 1341.11, 0.04
]
data['241_1341.11_0.10'] = [
'27_10000_46_10', 'R1rho', 799777399.1, 241.458629241, 1341.11, 0.1
]
data['241_1341.11_0.20'] = [
'30_10000_46_20', 'R1rho', 799777399.1, 241.458629241, 1341.11, 0.2
]
data['241_1341.11_0.40'] = [
'28_10000_46_40', 'R1rho', 799777399.1, 241.458629241, 1341.11, 0.4
]
# Check the number of time counts.
print("Checking the id return experiment.")
for id in cdp.exp_type:
exp_type = cdp.exp_type[id]
frq = cdp.spectrometer_frq[id]
offset = cdp.spin_lock_offset[id]
point = cdp.spin_lock_nu1[id]
# Loop over time
for time in loop_time(exp_type=exp_type,
frq=frq,
offset=offset,
point=point):
ids = find_intensity_keys(exp_type=exp_type,
frq=frq,
offset=offset,
point=point,
time=time)
print(exp_type, frq, offset, point, time,
data["%3.0f_%3.2f_%1.2f" % (offset, point, time)][5], id,
ids)
# Test the id return
self.assertEqual(len(ids), 1)
# Test the time point
self.assertEqual(
time, data["%3.0f_%3.2f_%1.2f" % (offset, point, time)][5])
self.assertEqual(
ids[0],
data["%3.0f_%3.2f_%1.2f" % (offset, point, time)][0])
def test_count_relax_times_r1rho(self):
"""Unit test of the count_relax_times() function.
This uses the data of the saved state attached to U{bug #21344<https://gna.org/bugs/?21344>}.
"""
# Load the state.
statefile = status.install_path + sep+'test_suite'+sep+'shared_data'+sep+'dispersion'+sep+'bug_21344_trunc.bz2'
state.load_state(statefile, force=True)
# Original data (spectrum id: exp_type, frq, omega_rf_ppm, spin_lock_field_strength, time_spin_lock).
data = dict()
data['46_0_35_0'] = ['R1rho', 799777399.1, 118.078, 431.0, 0.0]
data['48_0_35_4'] = ['R1rho', 799777399.1, 118.078, 431.0, 0.04]
data['47_0_35_10'] = ['R1rho', 799777399.1, 118.078, 431.0, 0.1]
data['49_0_35_20'] = ['R1rho', 799777399.1, 118.078, 431.0, 0.2]
data['36_0_39_0'] = ['R1rho', 799777399.1, 118.078, 651.2, 0.0]
data['39_0_39_4'] = ['R1rho', 799777399.1, 118.078, 651.2, 0.04]
data['37_0_39_10'] = ['R1rho', 799777399.1, 118.078, 651.2, 0.1]
data['40_0_39_20'] = ['R1rho', 799777399.1, 118.078, 651.2, 0.2]
data['38_0_39_40'] = ['R1rho', 799777399.1, 118.078, 651.2, 0.4]
data['41_0_41_0'] = ['R1rho', 799777399.1, 118.078, 800.5, 0.0]
data['44_0_41_4'] = ['R1rho', 799777399.1, 118.078, 800.5, 0.04]
data['42_0_41_10'] = ['R1rho', 799777399.1, 118.078, 800.5, 0.1]
data['45_0_41_20'] = ['R1rho', 799777399.1, 118.078, 800.5, 0.2]
data['43_0_41_40'] = ['R1rho', 799777399.1, 118.078, 800.5, 0.4]
data['31_0_43_0'] = ['R1rho', 799777399.1, 118.078, 984.0, 0.0]
data['34_0_43_4'] = ['R1rho', 799777399.1, 118.078, 984.0, 0.04]
data['32_0_43_10'] = ['R1rho', 799777399.1, 118.078, 984.0, 0.1]
data['35_0_43_20'] = ['R1rho', 799777399.1, 118.078, 984.0, 0.2]
data['33_0_43_40'] = ['R1rho', 799777399.1, 118.078, 984.0, 0.4]
data['1_0_46_0'] = ['R1rho', 799777399.1, 118.078, 1341.11, 0.0]
data['4_0_46_4'] = ['R1rho', 799777399.1, 118.078, 1341.11, 0.04]
data['2_0_46_10'] = ['R1rho', 799777399.1, 118.078, 1341.11, 0.1]
data['5_0_46_20'] = ['R1rho', 799777399.1, 118.078, 1341.11, 0.2]
data['3_0_46_40'] = ['R1rho', 799777399.1, 118.078, 1341.11, 0.4]
data['60_0_48_0'] = ['R1rho', 799777399.1, 118.078, 1648.5, 0.0]
data['63_0_48_4'] = ['R1rho', 799777399.1, 118.078, 1648.5, 0.04]
data['61_0_48_10'] = ['R1rho', 799777399.1, 118.078, 1648.5, 0.1]
data['62_0_48_14'] = ['R1rho', 799777399.1, 118.078, 1648.5, 0.14]
data['64_0_48_20'] = ['R1rho', 799777399.1, 118.078, 1648.5, 0.2]
data['11_500_46_0'] = ['R1rho', 799777399.1, 124.24703146206046, 1341.11, 0.0]
data['14_500_46_4'] = ['R1rho', 799777399.1, 124.24703146206046, 1341.11, 0.04]
data['12_500_46_10'] = ['R1rho', 799777399.1, 124.24703146206046, 1341.11, 0.1]
data['15_500_46_20'] = ['R1rho', 799777399.1, 124.24703146206046, 1341.11, 0.2]
data['13_500_46_40'] = ['R1rho', 799777399.1, 124.24703146206046, 1341.11, 0.4]
data['50_1000_41_0'] = ['R1rho', 799777399.1, 130.41606292412092, 800.5, 0.0]
data['53_1000_41_4'] = ['R1rho', 799777399.1, 130.41606292412092, 800.5, 0.04]
data['51_1000_41_10'] = ['R1rho', 799777399.1, 130.41606292412092, 800.5, 0.1]
data['54_1000_41_20'] = ['R1rho', 799777399.1, 130.41606292412092, 800.5, 0.2]
data['52_1000_41_40'] = ['R1rho', 799777399.1, 130.41606292412092, 800.5, 0.4]
data['21_1000_46_0'] = ['R1rho', 799777399.1, 130.41606292412092, 1341.11, 0.0]
data['24_1000_46_4'] = ['R1rho', 799777399.1, 130.41606292412092, 1341.11, 0.04]
data['22_1000_46_10'] = ['R1rho', 799777399.1, 130.41606292412092, 1341.11, 0.1]
data['25_1000_46_20'] = ['R1rho', 799777399.1, 130.41606292412092, 1341.11, 0.2]
data['23_1000_46_40'] = ['R1rho', 799777399.1, 130.41606292412092, 1341.11, 0.4]
data['65_1000_48_0'] = ['R1rho', 799777399.1, 130.41606292412092, 1648.5, 0.0]
data['68_1000_48_4'] = ['R1rho', 799777399.1, 130.41606292412092, 1648.5, 0.04]
data['66_1000_48_10'] = ['R1rho', 799777399.1, 130.41606292412092, 1648.5, 0.1]
data['67_1000_48_14'] = ['R1rho', 799777399.1, 130.41606292412092, 1648.5, 0.14]
data['69_1000_48_20'] = ['R1rho', 799777399.1, 130.41606292412092, 1648.5, 0.2]
data['55_2000_41_0'] = ['R1rho', 799777399.1, 142.75412584824184, 800.5, 0.0]
data['58_2000_41_4'] = ['R1rho', 799777399.1, 142.75412584824184, 800.5, 0.04]
data['56_2000_41_10'] = ['R1rho', 799777399.1, 142.75412584824184, 800.5, 0.1]
data['59_2000_41_20'] = ['R1rho', 799777399.1, 142.75412584824184, 800.5, 0.2]
data['57_2000_41_40'] = ['R1rho', 799777399.1, 142.75412584824184, 800.5, 0.4]
data['6_2000_46_0'] = ['R1rho', 799777399.1, 142.75412584824184, 1341.11, 0.0]
data['9_2000_46_4'] = ['R1rho', 799777399.1, 142.75412584824184, 1341.11, 0.04]
data['7_2000_46_10'] = ['R1rho', 799777399.1, 142.75412584824184, 1341.11, 0.1]
data['10_2000_46_20'] = ['R1rho', 799777399.1, 142.75412584824184, 1341.11, 0.2]
data['8_2000_46_40'] = ['R1rho', 799777399.1, 142.75412584824184, 1341.11, 0.4]
data['16_5000_46_0'] = ['R1rho', 799777399.1, 179.76831462060457, 1341.11, 0.0]
data['19_5000_46_4'] = ['R1rho', 799777399.1, 179.76831462060457, 1341.11, 0.04]
data['17_5000_46_10'] = ['R1rho', 799777399.1, 179.76831462060457, 1341.11, 0.1]
data['20_5000_46_20'] = ['R1rho', 799777399.1, 179.76831462060457, 1341.11, 0.2]
data['18_5000_46_40'] = ['R1rho', 799777399.1, 179.76831462060457, 1341.11, 0.4]
data['26_10000_46_0'] = ['R1rho', 799777399.1, 241.45862924120914, 1341.11, 0.0]
data['29_10000_46_4'] = ['R1rho', 799777399.1, 241.45862924120914, 1341.11, 0.04]
data['27_10000_46_10'] = ['R1rho', 799777399.1, 241.45862924120914, 1341.11, 0.1]
data['30_10000_46_20'] = ['R1rho', 799777399.1, 241.45862924120914, 1341.11, 0.2]
data['28_10000_46_40'] = ['R1rho', 799777399.1, 241.45862924120914, 1341.11, 0.4]
time_comp = {
'118.078_431.00':4,
'118.078_651.20':5,
'118.078_800.50':5,
'118.078_984.00':5,
'118.078_1341.11':5,
'118.078_1648.50':5,
'124.247_1341.11':5,
'130.416_800.50':5,
'130.416_1341.11':5,
'130.416_1648.50':5,
'142.754_800.50':5,
'142.754_1341.11':5,
'179.768_1341.11':5,
'241.459_1341.11':5}
# Check the number of time counts.
print("Checking the number of time counts.")
for id in cdp.exp_type:
exp_type = cdp.exp_type[id]
frq = cdp.spectrometer_frq[id]
offset = cdp.spin_lock_offset[id]
point = cdp.spin_lock_nu1[id]
count = count_relax_times(exp_type = exp_type, frq = frq, offset=offset, point = point, ei = cdp.exp_type_list.index(cdp.exp_type[id]))
print(id, exp_type, frq, offset, point, count)
# Test the time count
print(time_comp)
self.assertEqual(count, time_comp['%.3f_%.2f'%(offset, point)])
def test_find_intensity_keys_r1rho(self):
"""Unit test of the find_intensity_keys() function.
This uses the data of the saved state attached to U{bug #21344<https://gna.org/bugs/?21344>}.
"""
# Load the state.
statefile = status.install_path + sep+'test_suite'+sep+'shared_data'+sep+'dispersion'+sep+'bug_21344_trunc.bz2'
state.load_state(statefile, force=True)
# Original data (spectrum id: exp_type, frq, omega_rf_ppm, spin_lock_field_strength, time_spin_lock).
data = dict()
data['118_431.00_0.00'] = ['46_0_35_0', 'R1rho', 799777399.1, 118.078, 431.0, 0.0]
data['118_431.00_0.04'] = ['48_0_35_4', 'R1rho', 799777399.1, 118.078, 431.0, 0.04]
data['118_431.00_0.10'] = ['47_0_35_10', 'R1rho', 799777399.1, 118.078, 431.0, 0.1]
data['118_431.00_0.20'] = ['49_0_35_20', 'R1rho', 799777399.1, 118.078, 431.0, 0.2]
data['118_651.20_0.00'] = ['36_0_39_0', 'R1rho', 799777399.1, 118.078, 651.2, 0.0]
data['118_651.20_0.04'] = ['39_0_39_4', 'R1rho', 799777399.1, 118.078, 651.2, 0.04]
data['118_651.20_0.10'] = ['37_0_39_10', 'R1rho', 799777399.1, 118.078, 651.2, 0.1]
data['118_651.20_0.20'] = ['40_0_39_20', 'R1rho', 799777399.1, 118.078, 651.2, 0.2]
data['118_651.20_0.40'] = ['38_0_39_40', 'R1rho', 799777399.1, 118.078, 651.2, 0.4]
data['118_800.50_0.00'] = ['41_0_41_0', 'R1rho', 799777399.1, 118.078, 800.5, 0.0]
data['118_800.50_0.04'] = ['44_0_41_4', 'R1rho', 799777399.1, 118.078, 800.5, 0.04]
data['118_800.50_0.10'] = ['42_0_41_10', 'R1rho', 799777399.1, 118.078, 800.5, 0.1]
data['118_800.50_0.20'] = ['45_0_41_20', 'R1rho', 799777399.1, 118.078, 800.5, 0.2]
data['118_800.50_0.40'] = ['43_0_41_40', 'R1rho', 799777399.1, 118.078, 800.5, 0.4]
data['118_984.00_0.00'] = ['31_0_43_0', 'R1rho', 799777399.1, 118.078, 984.0, 0.0]
data['118_984.00_0.04'] = ['34_0_43_4', 'R1rho', 799777399.1, 118.078, 984.0, 0.04]
data['118_984.00_0.10'] = ['32_0_43_10', 'R1rho', 799777399.1, 118.078, 984.0, 0.1]
data['118_984.00_0.20'] = ['35_0_43_20', 'R1rho', 799777399.1, 118.078, 984.0, 0.2]
data['118_984.00_0.40'] = ['33_0_43_40', 'R1rho', 799777399.1, 118.078, 984.0, 0.4]
data['118_1341.11_0.00'] = ['1_0_46_0', 'R1rho', 799777399.1, 118.078, 1341.11, 0.0]
data['118_1341.11_0.04'] = ['4_0_46_4', 'R1rho', 799777399.1, 118.078, 1341.11, 0.04]
data['118_1341.11_0.10'] = ['2_0_46_10', 'R1rho', 799777399.1, 118.078, 1341.11, 0.1]
data['118_1341.11_0.20'] = ['5_0_46_20', 'R1rho', 799777399.1, 118.078, 1341.11, 0.2]
data['118_1341.11_0.40'] = ['3_0_46_40', 'R1rho', 799777399.1, 118.078, 1341.11, 0.4]
data['118_1648.50_0.00'] = ['60_0_48_0', 'R1rho', 799777399.1, 118.078, 1648.5, 0.0]
data['118_1648.50_0.04'] = ['63_0_48_4', 'R1rho', 799777399.1, 118.078, 1648.5, 0.04]
data['118_1648.50_0.10'] = ['61_0_48_10', 'R1rho', 799777399.1, 118.078, 1648.5, 0.1]
data['118_1648.50_0.14'] = ['62_0_48_14', 'R1rho', 799777399.1, 118.078, 1648.5, 0.14]
data['118_1648.50_0.20'] = ['64_0_48_20', 'R1rho', 799777399.1, 118.078, 1648.5, 0.2]
data['124_1341.11_0.00'] = ['11_500_46_0', 'R1rho', 799777399.1, 124.247031462, 1341.11, 0.0]
data['124_1341.11_0.04'] = ['14_500_46_4', 'R1rho', 799777399.1, 124.247031462, 1341.11, 0.04]
data['124_1341.11_0.10'] = ['12_500_46_10', 'R1rho', 799777399.1, 124.247031462, 1341.11, 0.1]
data['124_1341.11_0.20'] = ['15_500_46_20', 'R1rho', 799777399.1, 124.247031462, 1341.11, 0.2]
data['124_1341.11_0.40'] = ['13_500_46_40', 'R1rho', 799777399.1, 124.247031462, 1341.11, 0.4]
data['130_800.50_0.00'] = ['50_1000_41_0', 'R1rho', 799777399.1, 130.416062924, 800.5, 0.0]
data['130_800.50_0.04'] = ['53_1000_41_4', 'R1rho', 799777399.1, 130.416062924, 800.5, 0.04]
data['130_800.50_0.10'] = ['51_1000_41_10', 'R1rho', 799777399.1, 130.416062924, 800.5, 0.1]
data['130_800.50_0.20'] = ['54_1000_41_20', 'R1rho', 799777399.1, 130.416062924, 800.5, 0.2]
data['130_800.50_0.40'] = ['52_1000_41_40', 'R1rho', 799777399.1, 130.416062924, 800.5, 0.4]
data['130_1341.11_0.00'] = ['21_1000_46_0', 'R1rho', 799777399.1, 130.416062924, 1341.11, 0.0]
data['130_1341.11_0.04'] = ['24_1000_46_4', 'R1rho', 799777399.1, 130.416062924, 1341.11, 0.04]
data['130_1341.11_0.10'] = ['22_1000_46_10', 'R1rho', 799777399.1, 130.416062924, 1341.11, 0.1]
data['130_1341.11_0.20'] = ['25_1000_46_20', 'R1rho', 799777399.1, 130.416062924, 1341.11, 0.2]
data['130_1341.11_0.40'] = ['23_1000_46_40', 'R1rho', 799777399.1, 130.416062924, 1341.11, 0.4]
data['130_1648.50_0.00'] = ['65_1000_48_0', 'R1rho', 799777399.1, 130.416062924, 1648.5, 0.0]
data['130_1648.50_0.04'] = ['68_1000_48_4', 'R1rho', 799777399.1, 130.416062924, 1648.5, 0.04]
data['130_1648.50_0.10'] = ['66_1000_48_10', 'R1rho', 799777399.1, 130.416062924, 1648.5, 0.1]
data['130_1648.50_0.14'] = ['67_1000_48_14', 'R1rho', 799777399.1, 130.416062924, 1648.5, 0.14]
data['130_1648.50_0.20'] = ['69_1000_48_20', 'R1rho', 799777399.1, 130.416062924, 1648.5, 0.2]
data['143_800.50_0.00'] = ['55_2000_41_0', 'R1rho', 799777399.1, 142.754125848, 800.5, 0.0]
data['143_800.50_0.04'] = ['58_2000_41_4', 'R1rho', 799777399.1, 142.754125848, 800.5, 0.04]
data['143_800.50_0.10'] = ['56_2000_41_10', 'R1rho', 799777399.1, 142.754125848, 800.5, 0.1]
data['143_800.50_0.20'] = ['59_2000_41_20', 'R1rho', 799777399.1, 142.754125848, 800.5, 0.2]
data['143_800.50_0.40'] = ['57_2000_41_40', 'R1rho', 799777399.1, 142.754125848, 800.5, 0.4]
data['143_1341.11_0.00'] = ['6_2000_46_0', 'R1rho', 799777399.1, 142.754125848, 1341.11, 0.0]
data['143_1341.11_0.04'] = ['9_2000_46_4', 'R1rho', 799777399.1, 142.754125848, 1341.11, 0.04]
data['143_1341.11_0.10'] = ['7_2000_46_10', 'R1rho', 799777399.1, 142.754125848, 1341.11, 0.1]
data['143_1341.11_0.20'] = ['10_2000_46_20', 'R1rho', 799777399.1, 142.754125848, 1341.11, 0.2]
data['143_1341.11_0.40'] = ['8_2000_46_40', 'R1rho', 799777399.1, 142.754125848, 1341.11, 0.4]
data['180_1341.11_0.00'] = ['16_5000_46_0', 'R1rho', 799777399.1, 179.768314621, 1341.11, 0.0]
data['180_1341.11_0.04'] = ['19_5000_46_4', 'R1rho', 799777399.1, 179.768314621, 1341.11, 0.04]
data['180_1341.11_0.10'] = ['17_5000_46_10', 'R1rho', 799777399.1, 179.768314621, 1341.11, 0.1]
data['180_1341.11_0.20'] = ['20_5000_46_20', 'R1rho', 799777399.1, 179.768314621, 1341.11, 0.2]
data['180_1341.11_0.40'] = ['18_5000_46_40', 'R1rho', 799777399.1, 179.768314621, 1341.11, 0.4]
data['241_1341.11_0.00'] = ['26_10000_46_0', 'R1rho', 799777399.1, 241.458629241, 1341.11, 0.0]
data['241_1341.11_0.04'] = ['29_10000_46_4', 'R1rho', 799777399.1, 241.458629241, 1341.11, 0.04]
data['241_1341.11_0.10'] = ['27_10000_46_10', 'R1rho', 799777399.1, 241.458629241, 1341.11, 0.1]
data['241_1341.11_0.20'] = ['30_10000_46_20', 'R1rho', 799777399.1, 241.458629241, 1341.11, 0.2]
data['241_1341.11_0.40'] = ['28_10000_46_40', 'R1rho', 799777399.1, 241.458629241, 1341.11, 0.4]
# Check the number of time counts.
print("Checking the id return experiment.")
for id in cdp.exp_type:
exp_type = cdp.exp_type[id]
frq = cdp.spectrometer_frq[id]
offset = cdp.spin_lock_offset[id]
point = cdp.spin_lock_nu1[id]
# Loop over time
for time in loop_time(exp_type=exp_type, frq=frq, offset=offset, point=point):
ids = find_intensity_keys(exp_type=exp_type, frq=frq, offset=offset, point=point, time=time)
print(exp_type, frq, offset, point, time, data["%3.0f_%3.2f_%1.2f"%(offset, point, time)][5], id, ids)
# Test the id return
self.assertEqual(len(ids), 1)
# Test the time point
self.assertEqual(time, data["%3.0f_%3.2f_%1.2f"%(offset, point, time)][5])
self.assertEqual(ids[0], data["%3.0f_%3.2f_%1.2f"%(offset, point, time)][0])
def test_loop_exp_frq_offset_point_time_cpmg(self):
"""Unit test of the loop_exp_frq_offset_point_time() function.
This uses the data of the saved state attached to U{bug #21665<https://web.archive.org/web/https://gna.org/bugs/?21665>}.
"""
# Load the state.
statefile = status.install_path + sep + 'test_suite' + sep + 'shared_data' + sep + 'dispersion' + sep + 'bug_21665.bz2'
state.load_state(statefile, force=True)
# Original data (exp_type, frq, offset, point).
data = [[
'SQ CPMG', 499862140.0, 0,
[
50.0, 100.0, 150.0, 200.0, 250.0, 300.0, 350.0, 400.0, 450.0,
500.0, 550.0, 600.0, 650.0, 700.0, 800.0, 900.0, 1000.0
], 0.04
],
[
'SQ CPMG', 599890858.69999993, 0,
[
33.3333, 66.666, 100.0, 133.333, 166.666, 200.0,
233.333, 266.666, 300.0, 333.333, 366.666, 400.0,
466.666, 533.333, 666.666, 866.666, 1000.0
], 0.06
]]
# Original indices (ei, mi, oi).
indices = [
[
0, 0, 0,
[0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16], 0
],
[
0, 1, 0,
[0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16], 0
]
]
# Check the number of iterations.
print("Checking the number of iterations of the loop.")
count = 0
for exp_type, frq, offset, point, time, ei, mi, oi, di, ti in loop_exp_frq_offset_point_time(
return_indices=True):
print(exp_type, frq, offset, point, time, ei, mi, oi, di, ti)
count += 1
self.assertEqual(count, 34)
# Check the values.
print("Checking the values returned by the loop.")
frq_index = 0
disp_index = 0
for exp_type, frq, offset, point, time, ei, mi, oi, di, ti in loop_exp_frq_offset_point_time(
return_indices=True):
# Check the experiment info.
self.assertEqual(exp_type, data[frq_index][0])
self.assertEqual(ei, indices[frq_index][0])
# Check the frequency info.
self.assertEqual(frq, data[frq_index][1])
self.assertEqual(mi, indices[frq_index][1])
# Check the offset info.
self.assertEqual(offset, data[frq_index][2])
self.assertEqual(oi, indices[frq_index][2])
# Check the dispersion point info.
self.assertAlmostEqual(point, data[frq_index][3][disp_index], 2)
self.assertEqual(di, indices[frq_index][3][disp_index])
# Check the time point info.
self.assertEqual(time, data[frq_index][4])
self.assertEqual(ti, indices[frq_index][4])
# Increment the data index.
if disp_index == 16:
frq_index += 1
disp_index = 0
else:
disp_index += 1
