from rk4 import rk4
import numpy as np

def diff(x):
    return x

def mercury(u0, phi0, dphi):
    x = np.array([u0, phi0])
dt = 2
x = rk4(diff, x, dt)
return x
mercury(1, 1, 2)
Defining a global function in a Python script