def test_each_element():
freqs = [0.001, 1.0, 1000]
correct_vals = {'R': [0.1, 0.1, 0.1],
'C': [-1591.5494309189532j,
-1.5915494309189535j,
-0.0015915494309189533j],
'L': [0.000628319j, 0.628319j, 628.319j],
'W': [(1.26156626-1.26156626j),
(0.03989423-0.03989423j),
(0.00126157-0.00126157j)],
'Wo': [(0.033333332999112786-79.57747433847442j),
(0.03300437046673635-0.08232866785870396j),
(0.0019947114020071634-0.0019947114020071634j)],
'Ws': [(0.09999998-4.18878913e-05j),
(0.08327519-3.33838167e-02j),
(0.00199471-1.99471140e-03j)],
'CPE': [(26.216236841407248-8.51817171087997j),
(6.585220960717244-2.139667994182814j),
(1.6541327179718126-0.537460300252313j)],
'La': [(0.21769191+0.07073239j),
(0.86664712+0.28159072j),
(3.45018434+1.12103285j)],
'G': [(0.09999994078244179-0.00006283179105931961j),
(0.07107755021941357-0.03427465211788068j),
(0.00199550459845528-0.0019939172581851707j)],
'Gs': [(0.3432733166533134-0.00041895248193532704j),
(0.1391819314527732-0.16248466787637972j),
(0.0019955029598887875-0.0019939170758457437j)],
'TLMQ': [(20.42237173-10.38874276j),
(2.6000937-1.30789949j),
(0.35594139-0.16491599j)],
'T': [(1.00041-0.00837309j),
(0.0156037-0.114062j),
(0.00141056-0.00141039j)],
'K': [(0.099999842086579 - 0.000125663507704j),
(0.038772663673915 - 0.048723166143232j),
(6.332569967499333e-08 - 7.957742115295703e-05j)]}
input_vals = [0.1, 0.2, 0.3, 0.4]
for key, f in circuit_elements.items():
# don't test the outputs of series and parallel functions
if key not in ['s', 'p']:
num_inputs = f.num_params
val = f(input_vals[:num_inputs], freqs)
assert np.isclose(val, correct_vals[key]).all()
# check for typing:
with pytest.raises(AssertionError):
f = circuit_elements['R']
f(1, 2)
# test for handling more wrong inputs
with pytest.raises(AssertionError):
f(['hi'], ['yes', 'hello'])
# Test no overflow in T at high frequencies
with pytest.warns(None) as record:
circuit_elements['T']([1, 2, 50, 100], [10000])
assert not record
def test_all():
freqs = [0.001, 1.0, 1000]
correct_vals = {
'R': [0.1, 0.1, 0.1],
'C':
[-1591.5494309189532j, -1.5915494309189535j, -0.0015915494309189533j],
'L': [0.000628319j, 0.628319j, 628.319j],
'W': [(1.26156626 - 1.26156626j), (0.03989423 - 0.03989423j),
(0.00126157 - 0.00126157j)],
'Wo': [(0.033333332999112786 - 79.57747433847442j),
(0.03300437046673635 - 0.08232866785870396j),
(0.0019947114020071634 - 0.0019947114020071634j)],
'Ws': [(0.09999998 - 4.18878913e-05j), (0.08327519 - 3.33838167e-02j),
(0.00199471 - 1.99471140e-03j)],
'CPE': [(26.216236841407248 - 8.51817171087997j),
(6.585220960717244 - 2.139667994182814j),
(1.6541327179718126 - 0.537460300252313j)],
'G': [(0.09999994078244179 - 0.00006283179105931961j),
(0.07107755021941357 - 0.03427465211788068j),
(0.00199550459845528 - 0.0019939172581851707j)],
'Gs': [(0.3432733166533134 - 0.00041895248193532704j),
(0.1391819314527732 - 0.16248466787637972j),
(0.0019955029598887875 - 0.0019939170758457437j)],
'T': [(1.00041 - 0.00837309j), (0.0156037 - 0.114062j),
(0.00141056 - 0.00141039j)],
'K': [(0.099999842086579 - 0.000125663507704j),
(0.038772663673915 - 0.048723166143232j),
(6.332569967499333e-08 - 7.957742115295703e-05j)]
}
input_vals = [0.1, 0.2, 0.3, 0.4]
for key, f in circuit_elements.items():
# don't test the outputs of series and parallel functions
if key not in ['s', 'p']:
num_inputs = f.num_params
val = f(input_vals[:num_inputs], freqs)
assert np.isclose(val, correct_vals[key]).all()
# check for typing:
try:
f = circuit_elements['R']
f(1, 2)
except (AssertionError):
pass
else:
raise Exception('unhandled error occurred')
# test for handling more wrong inputs
try:
f(['hi'], ['yes', 'hello'])
except (AssertionError):
pass
else:
raise Exception('unhandled error occurred')