import matplotlib.pyplot as plt
import neuromodels as nm

hh = nm.HodgkinHuxley(V_rest=-70)
T = 50.
dt = 0.025

def stimulus(t):
    return 10 if 10 <= t <= 40 else 0

hh.solve(stimulus, T, dt)
t = hh.t
V = hh.V

plt.plot(t, V)
plt.xlabel('Time [ms]')
plt.ylabel('Membrane potential [mV]')
plt.show()