def test_return_fit_returns_correct_ellipse(n_points): X = make_dataset(center=[0, 0], width=1, height=.5, phi=0, n_points=n_points) elp = LsqEllipse().fit(X) x = elp.return_fit(n_points) nptest.assert_array_almost_equal(x, X)
def test_return_fit_returns_correct_ellipse(n_points): X = make_dataset(center=[0, 0], width=1, height=.5, phi=0, n_points=n_points) elp = LsqEllipse().fit(X) t = np.linspace(0, 1.8 * np.pi, n_points) center, major, minor, phi = elp.as_parameters() with mock.patch.object(LsqEllipse, 'as_parameters') as as_parameters: as_parameters.return_value = (center, *_normalize_result(major, minor, phi)) x = elp.return_fit(n_points, t=t) nptest.assert_array_almost_equal(x, X)