def test_lead_field_class_same_output_as_function():
global old_code
lf = Lead_Field()
for i in range(len(old_code)):
lead_field_class = lf.calculate(old_code[i]['gen_conf'])
lead_field_function = calculate_lead_field(old_code[i]['gen_conf'])
assert_array_equal(lead_field_class, lead_field_function)
def test_random_generator_placement_good_for_lead_field_calculation():
gen_conf = random_generator_placement()
calculate_lead_field(gen_conf)
from numpy import pi
from simulation import random_generator_configuration
from lead_field import calculate_lead_field
descriptions = []
gen_confs = []
lead_fields = []
descriptions.append('A single dipole, located at top of sphere, oriented ' +\
'parallel to the surface')
gen_confs.append([{'depth': 6, 'theta': 0, 'phi': 0, 'orientation': 0,
'orientation_phi': 0, 'magnitude': 1}])
lead_fields.append(calculate_lead_field(gen_confs[-1]))
descriptions.append('A single dipole, located at top of sphere, oriented ' +\
'perpendicular to the surface')
gen_confs.append([{'depth': 5, 'theta': 0, 'phi': 0, 'orientation': pi/2,
'orientation_phi': 0, 'magnitude': 1}])
lead_fields.append(calculate_lead_field(gen_confs[-1]))
descriptions.append('A single dipole, at a specific location')
gen_confs.append([{'depth': 7, 'theta': 3*pi/8, 'phi': 3*pi/4,
'orientation': pi/4, 'orientation_phi': 3*pi/4,
'magnitude': 1}])
lead_fields.append(calculate_lead_field(gen_confs[-1]))
descriptions.append('5 generators in a random configuration (with specified' +\
'seed')
def test_lead_field_same_output_as_old_code_3():
global old_code
lead_field = calculate_lead_field(old_code[3]['gen_conf'])
assert_array_almost_equal(lead_field, old_code[3]['lead_field'], 17)
