def calcNetForceExertedByXY(self):
print("Checking setNetForce...")
p1 = Planet(1.0, 1.0, 3.0, 4.0, 5.0, "jupiter.gif")
p2 = Planet(2.0, 1.0, 3.0, 4.0, 4e11, "jupiter.gif")
p3 = Planet(4.0, 5.0, 3.0, 4.0, 5.0, "jupiter.gif")
p4 = Planet(3.0, 2.0, 3.0, 4.0, 5.0, "jupiter.gif")
planets = [p2, p3, p4]
xNetForce = p1.calcNetForceExertedByX(planets)
yNetForce = p1.calcNetForceExertedByY(planets)
print(xNetForce, yNetForce)
self.checkEquals(133.4, self.round(xNetForce, '.01'),
"calcNetForceExertedByX()")
self.checkEquals(0.0, self.round(yNetForce, '.01'),
"calcNetForceExertedByY()")
print(
"Running test again, but with array that contains the target planet."
)
planets = [p1, p2, p3, p4]
xNetForce = p1.calcNetForceExertedByX(planets)
yNetForce = p1.calcNetForceExertedByY(planets)
self.checkEquals(133.4, self.round(xNetForce, '.01'),
"calcNetForceExertedByX()")
self.checkEquals(0.0, self.round(yNetForce, '.01'),
"calcNetForceExertedByY()")
