add test for circle_fitting

This commit is contained in:
Atsushi Sakai
2018-05-12 22:14:44 +09:00
parent 17b87c0822
commit 3c4d1c457a
2 changed files with 24 additions and 8 deletions

View File

@@ -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()

View File

@@ -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()