import numpy as np
import matplotlib.pyplot as plt
%matplotlib inline
# Declare constants
# in MKS units
G = 6.6738E-11
c = 2.9979E8
Msol = 1.9891E30
Rsol = 6.955E8
def rschwarz(mass):
return G*mass/c**2
def gamma(velocity):
return 1./np.sqrt(1 - (v/c)**2)
def grav_dilation(mass, radius):
return np.sqrt(1 - G*mass/(radius*c**2))
masses = np.linspace(1,100,100)
plt.plot(masses, rschwarz(masses*Msol)/Rsol)
plt.show()