diff --git a/Mapping/circle_fitting/circle_fitting.py b/Mapping/circle_fitting/circle_fitting.py index 74eced20..b9574081 100644 --- a/Mapping/circle_fitting/circle_fitting.py +++ b/Mapping/circle_fitting/circle_fitting.py @@ -12,6 +12,8 @@ import math 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 new file mode 100644 index 00000000..d1845482 --- /dev/null +++ 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()