def test_read_command_line_argparse():
I = 1.6; a = 1.8; T = 2.2; theta = 0.5
dt_values = [0.1, 0.2, 0.05]
# Expected return from read_command_line_positional
expected = [I, a, T, theta, dt_values]
# Construct corresponding sys.argv array
cml = '%s --a %s --I %s --T %s --scheme CN --dt ' % \
(sys.argv[0], a, I, T)
cml = cml + ' '.join([str(dt) for dt in dt_values])
sys.argv = cml.split()
computed = decay.read_command_line_argparse()
for expected_arg, computed_arg in zip(expected, computed):
assert expected_arg == computed_arg
def main():
# Read parameters, solve and plot
I, a, T, theta, dt_values = read_command_line_argparse()
dt = dt_values[0] # use only the first dt value
u_scaled, t_scaled = solver_scaled(T, dt, theta)
u, t = unscale(u_scaled, t_scaled, I, a)
import matplotlib.pyplot as plt
plt.figure()
plt.plot(t_scaled, u_scaled)
plt.xlabel('scaled time'); plt.ylabel('scaled velocity')
plt.title('Universial solution of scaled problem')
plt.savefig('tmp1.png'); plt.savefig('tmp1.pdf')
plt.figure()
plt.plot(t, u)
plt.xlabel('t'); plt.ylabel('u')
plt.title('I=%g, a=%g, theta=%g' % (I, a, theta))
plt.savefig('tmp.png')
plt.show()
def main():
# Read parameters, solve and plot
I, a, T, theta, dt_values = read_command_line_argparse()
dt = dt_values[0] # use only the first dt value
u_scaled, t_scaled = solver_scaled(T, dt, theta)
u, t = unscale(u_scaled, t_scaled, I, a)
import matplotlib.pyplot as plt
plt.figure()
plt.plot(t_scaled, u_scaled)
plt.xlabel('scaled time')
plt.ylabel('scaled velocity')
plt.title('Universial solution of scaled problem')
plt.savefig('tmp1.png')
plt.savefig('tmp1.pdf')
plt.figure()
plt.plot(t, u)
plt.xlabel('t')
plt.ylabel('u')
plt.title('I=%g, a=%g, theta=%g' % (I, a, theta))
plt.savefig('tmp.png')
plt.show()