from pylab import *
from numpy import *  #je dobre zvyknutis si aj na tento import
#uvedene dva importy v neinteraktivne spustanych .py suboroch robia nieco podobne
# ako ked sa interaktivny ipython spusti s option --pylab 

G=6.674e-11 #gravitacna konstanta
M=2.    # hmotnost centralneho telesa
m=1.	# hmotnost prenasaneho telesa
r1=array([1.,2.,3.]) 	#poloha startovacieho bodu presunu
r2=array([2.,5.,6.])	#poloha kocoveho bodu presunu
A=0.  # vynulovany sumator na vypocet prace

dr=?????  #maly vektorik v smere drahy prenosu do r1 k r2. CFelu usecku rozdelit na 1000 rovnakych kuskov

def sila(r):
	#funkcia, ktora ma vratit gravitacnu silu v mieste r
	abs0=sqrt(r[0]**2+r[1]**2+r[2]**2) #velkost vektora r, suradnice sa cisluju 0,1,2 v Pythone
	return(?????)  # vypocitat a vratit vektor gravitacnej sily v mieste r (od pociatku)

#teraz ide vypocet prace
for i in range(0,1000):
	r=r1+i*dr  #aktualna poloha prenasaneho bodu
	F=-sila(r) #sila, ktorou posobi trpsalik
	A=A+?????  # vypocitat a pricitat pracu trpaslika na useku dr

print(A)

#teraz pre kontrolu vypocet cez potencialnu energiu
abs2=sqrt(r2[0]**2+r2[1]**2+r2[2]**2) # vzdialenost koncoveho bodu od pociatku
abs1=sqrt(r1[0]**2+r1[1]**2+r1[2]**2) #vzdialenost pociatocneho bodu od pociatku
a=(????)-(????)  #vypocitat potencialne energuie v bodoch r2 a r1 a odcitat ich
print(a)