In [1]:

    

import numpy as np
import sympy as sp
from pyNT.ode import ODE, get_stability_region
from pyNT.ode import base_ODE


    

In [13]:

    

%matplotlib inline
import matplotlib.pyplot as plt

func_name = base_ODE().add_EE(step_list = range(1, 9))

print func_name

solver_list = ['Euler',
               'Heun',
               'cRK',
               func_name]

contours = get_stability_region(
    solver_list = solver_list)

fig = plt.figure(figsize=(10,10))
ax = fig.add_subplot(111)
for i in range(len(solver_list)):
    line = ax.plot(contours[i][0][:, 0],
            contours[i][0][:, 1],
            label = solver_list[i])
    for path in contours[i][1:]:
        ax.plot(path[:, 0],
                path[:, 1],
                color = line[0].get_color())
ax.legend(loc='best')
ax.set_aspect('equal')


    

    




EE_1_2_3_4_5_6_7_8






    

In [ ]: