diff --git a/PathPlanning/RRTStarReedsShepp/rrt_star_reeds_shepp.py b/PathPlanning/RRTStarReedsShepp/rrt_star_reeds_shepp.py index ccc0be21..ed8d74ae 100644 --- a/PathPlanning/RRTStarReedsShepp/rrt_star_reeds_shepp.py +++ b/PathPlanning/RRTStarReedsShepp/rrt_star_reeds_shepp.py @@ -300,11 +300,12 @@ def main(): path = rrt.Planning(animation=show_animation) # Draw final path - rrt.DrawGraph() - plt.plot([x for (x, y) in path], [y for (x, y) in path], '-r') - plt.grid(True) - plt.pause(0.001) - plt.show() + if show_animation: + rrt.DrawGraph() + plt.plot([x for (x, y) in path], [y for (x, y) in path], '-r') + plt.grid(True) + plt.pause(0.001) + plt.show() if __name__ == '__main__': diff --git a/tests/test_rrt_star_reeds_shepp.py b/tests/test_rrt_star_reeds_shepp.py new file mode 100644 index 00000000..611568bf --- /dev/null +++ b/tests/test_rrt_star_reeds_shepp.py @@ -0,0 +1,15 @@ +from unittest import TestCase + +import sys +sys.path.append("./PathPlanning/RRTStarReedsShepp/") + +from PathPlanning.RRTStarReedsShepp import rrt_star_reeds_shepp as m + +print(__file__) + + +class Test(TestCase): + + def test1(self): + m.show_animation = False + m.main()