add test

AerialNavigation/rocket_powered_landing/rocket_powered_landing.py

@@ -9,7 +9,7 @@ Ref:
 - Python implementation of 'Successive Convexification for 6-DoF Mars Rocket Powered Landing with Free-Final-Time' paper
 by Michael Szmuk and Behçet Açıkmeşe.
 
-- inspired by EmbersArc/SuccessiveConvexificationFreeFinalTime: Implementation of "Successive Convexification for 6-DoF Mars Rocket Powered Landing with Free-Final-Time" https://github.com/EmbersArc/SuccessiveConvexificationFreeFinalTime
+- EmbersArc/SuccessiveConvexificationFreeFinalTime: Implementation of "Successive Convexification for 6-DoF Mars Rocket Powered Landing with Free-Final-Time" https://github.com/EmbersArc/SuccessiveConvexificationFreeFinalTime
 
 """
 
@@ -35,6 +35,8 @@ W_NU = 1e5  # virtual control
 solver = 'ECOS'
 verbose_solver = False
 
+show_animation = True
+
 
 class Integrator:
     def __init__(self, m, K):
@@ -656,7 +658,8 @@ def main():
             print(f'Converged after {it + 1} iterations.')
             break
 
-    plot_animation(X, U)
+    if show_animation:
+        plot_animation(X, U)
 
     print("done!!")
 

tests/test_rocket_powered_landing.py

@@ -0,0 +1,14 @@
+from unittest import TestCase
+
+import sys
+sys.path.append("./AerialNavigation/rocket_powered_landing/")
+
+from AerialNavigation.rocket_powered_landing import rocket_powered_landing as m
+print(__file__)
+
+
+class Test(TestCase):
+
+    def test1(self):
+        m.show_animation = False
+        m.main()