def test_rf_lim(self, rf_lim, out):
model = ret.EllipseModel()
model.params = ret._make_ellipse_model_params_focus(47, 30, 10, 30, 0)
is_good = ret.is_ellipse_good(
ellipse_model=model, data=None, xf=50, yf=30, rf_lim=rf_lim
)
assert is_good == out
def test_yf(self, yf, out):
model = ret.EllipseModel()
model.params = ret._make_ellipse_model_params_focus(70, 10, 10, 30, 0)
is_good = ret.is_ellipse_good(
ellipse_model=model, data=None, xf=70, yf=yf, rf_lim=5
)
assert is_good == out
def test_different_rotations(self, r):
model = ret.EllipseModel()
model.params = ret._make_ellipse_model_params_focus(50, 30, 3, 6.5, r)
is_good = ret.is_ellipse_good(
ellipse_model=model, data=None, xf=50, yf=30, rf_lim=5
)
assert is_good
def test_simple(self):
params = ret._make_ellipse_model_params_focus(30, 50, 30, 20, 0)
model = ret.EllipseModel()
model.params = params
is_good = ret.is_ellipse_good(
ellipse_model=model, data=None, xf=30, yf=50, rf_lim=5
)
assert is_good
def test_semi_len_ratio_lim(self, semi_len_ratio_lim, out):
model = ret.EllipseModel()
model.params = ret._make_ellipse_model_params_focus(50, 30, 9, 4.5, 0)
is_good = ret.is_ellipse_good(
ellipse_model=model,
data=None,
xf=50,
yf=30,
rf_lim=5,
semi_len_ratio_lim=semi_len_ratio_lim,
)
assert is_good == out
def test_semi_len_min(self, semi_len_min, out):
model = ret.EllipseModel()
model.params = ret._make_ellipse_model_params_focus(50, 30, 6.6, 10, 0)
is_good = ret.is_ellipse_good(
ellipse_model=model,
data=None,
xf=50,
yf=30,
rf_lim=5,
semi_len_min=semi_len_min,
)
assert is_good == out
def test_many_different_ones(self, x, y, r, smin, smax, srl, out):
model = ret.EllipseModel()
model.params = ret._make_ellipse_model_params_focus(5, 3, 4, 6.5, 0.1)
is_good = ret.is_ellipse_good(
ellipse_model=model,
data=None,
xf=x,
yf=y,
rf_lim=r,
semi_len_min=smin,
semi_len_max=smax,
semi_len_ratio_lim=srl,
)
assert is_good == out
def test_semi_len_max(self, a, b, semi_len_max, out):
model = ret.EllipseModel()
model.params = ret._make_ellipse_model_params_focus(50, 30, a, b, 0)
params = list(model.params)
# Changing the params like this is necessary to test if b is larger
# than semi_len_max. This is due to the content in model.params not being
# the same as the input in ret._make_ellipse_model_params_focus.
params[2:5] = a, b, 0.0
model.params = tuple(params)
is_good = ret.is_ellipse_good(
ellipse_model=model,
data=None,
xf=50,
yf=30,
rf_lim=5,
semi_len_max=semi_len_max,
)
assert is_good == out
