test.py

import numpy as np
import matplotlib.pyplot as plt
from scipy.integrate import odeint



def Atwood(y,t,M,m,g):

    g=9.81
    M = int(input())
    m = int(input())
    #theta'(t)=omega(t)
    #radius'(t)=Rho(t)

    #omega'(t)=(1/radius(t))(-g*sin(theta(t))-2(Rho(t)*omega(t))
    #Rho'(t)=(1/(M+m))(m*radius(t)*(omega(t)^2)-M*g+m*g*cos(theta(t)))

    theta, omega, radius, Rho = y

    dydt = [omega, (1/radius)*(-g*(np.sin(theta))-2*(Rho*omega)), Rho, (1/(M+m))*(m*radius*(omega**2)-M*g+m*g*(np.cos(theta)))]

    return dydt

#y0=[theta, change in theta, radius, change in radius]
y0 = [0 ,2 ,1 ,1]
#time points
t = np.linspace(0,10,101)
#Solving the ODE
sul = odeint(Atwood, y0, t, args=(M,m,g))