add test for circle_fitting

abshk-jr · May 12, 2018 · 3c4d1c4 · 3c4d1c4
1 parent 17b87c0
commit 3c4d1c4
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 8 deletions.
diff --git a/Mapping/circle_fitting/circle_fitting.py b/Mapping/circle_fitting/circle_fitting.py
@@ -12,6 +12,8 @@
 import random
 import numpy as np
 
+show_animation = True
+
 
 def circle_fitting(x, y):
     """
@@ -120,15 +122,17 @@ def main():
         ex, ey, er, error = circle_fitting(x, y)
         print("Error:", error)
 
-        plt.cla()
-        plt.axis("equal")
-        plt.plot(0.0, 0.0, "*r")
-        plot_circle(cx, cy, cr)
-        plt.plot(x, y, "xr")
-        plot_circle(ex, ey, er, "-r")
-        plt.pause(dt)
+        if show_animation:
+            plt.cla()
+            plt.axis("equal")
+            plt.plot(0.0, 0.0, "*r")
+            plot_circle(cx, cy, cr)
+            plt.plot(x, y, "xr")
+            plot_circle(ex, ey, er, "-r")
+            plt.pause(dt)
+
+    print("Done")
 
 
 if __name__ == '__main__':
-    #  main1()
     main()
diff --git a/tests/test_circle_fitting.py b/tests/test_circle_fitting.py
@@ -0,0 +1,12 @@
+from unittest import TestCase
+
+from Mapping.circle_fitting import circle_fitting as m
+
+print(__file__)
+
+
+class Test(TestCase):
+
+    def test1(self):
+        m.show_animation = False
+        m.main()