solve_ivp code snippets in python