test.py

import numpy as np
from core_functions import *


masses = [5.972e24, 419725]

u_0 = np.array([[0,0,0,0,0,0], [400000 + 6378100, 0, 0, 0, 7823.6, 0]])

T = 5400

delta_t = 10

u, times = ivp_RK4(u_0, T, delta_t, masses)

rx = []
ry = []
rz = []

for i in range(len(times)):
    rx.append(u[i][1][0])
    ry.append(u[i][1][1])
    rz.append(u[i][1][2])

import matplotlib.pyplot as plt

plt.plot(rx,ry)

plt.show()