()
| 30 | |
| 31 | |
| 32 | def test_ellipse_fitter(): |
| 33 | fitter = trackingutils.EllipseFitter() |
| 34 | assert fitter.fit(np.random.rand(2, 2)) is None |
| 35 | xy = np.asarray([[-2, 0], [2, 0], [0, 1], [0, -1]], dtype=float) |
| 36 | assert fitter.fit(xy) is not None |
| 37 | fitter.sd = 0 |
| 38 | el = fitter.fit(xy) |
| 39 | assert np.isclose(el.parameters, [0, 0, 4, 2, 0]).all() |
| 40 | |
| 41 | |
| 42 | def test_ellipse_tracker(ellipse): |