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)