def test_r_min_and_r_max(self):
peak_list = np.array([[50, 0], [0, 30]])
peak_filtered_list0 = ct._filter_peak_list_radius(peak_list,
xc=0,
yc=0,
r_min=20,
r_max=40)
assert (peak_filtered_list0 == np.array([[0, 30]])).all()
peak_filtered_list1 = ct._filter_peak_list_radius(peak_list,
xc=0,
yc=0,
r_min=20,
r_max=60)
assert (peak_filtered_list1 == np.array([[50, 0], [0, 30]])).all()
peak_filtered_list2 = ct._filter_peak_list_radius(peak_list,
xc=0,
yc=0,
r_min=40,
r_max=60)
assert (peak_filtered_list2 == np.array([[50, 0]])).all()
peak_filtered_list3 = ct._filter_peak_list_radius(peak_list,
xc=0,
yc=0,
r_min=10,
r_max=20)
assert len(peak_filtered_list3) == 0
peak_filtered_list4 = ct._filter_peak_list_radius(peak_list,
xc=0,
yc=0,
r_min=90,
r_max=99)
assert len(peak_filtered_list4) == 0
def test_r_min_larger_than_r_max(self):
peak_list = np.array([[50, 10], [50, 50]])
with pytest.raises(ValueError):
ct._filter_peak_list_radius(peak_list,
xc=10,
yc=50,
r_min=50,
r_max=30)
def test_r_lim(self):
peak_list = np.array([[50, 50], [50, 30]])
peak_filtered_list0 = ct._filter_peak_list_radius(
peak_list, xc=50, yc=50, r_min=19
)
assert (peak_filtered_list0 == np.array([[50, 30]])).all()
peak_filtered_list1 = ct._filter_peak_list_radius(
peak_list, xc=50, yc=50, r_min=21
)
assert len(peak_filtered_list1) == 0
def test_simple(self):
peak_list = np.random.randint(100, size=(1000, 2))
peak_filtered_list0 = ct._filter_peak_list_radius(
peak_list, xc=50, yc=50, r_min=30
)
assert len(peak_list) != len(peak_filtered_list0)
peak_filtered_list1 = ct._filter_peak_list_radius(
peak_list, xc=50, yc=50, r_min=1000
)
assert len(peak_filtered_list1) == 0
def test_xc_yc(self):
peak_list = np.array([[50, 10], [50, 50]])
peak_filtered_list = ct._filter_peak_list_radius(peak_list,
xc=10,
yc=50,
r_min=10)
assert (peak_filtered_list == np.array([[50, 50]])).all()
def test_r_max(self):
peak_list = np.array([[50, 0], [0, 30]])
peak_filtered_list0 = ct._filter_peak_list_radius(
peak_list, xc=0, yc=0, r_max=40
)
assert (peak_filtered_list0 == np.array([[0, 30]])).all()
peak_filtered_list1 = ct._filter_peak_list_radius(
peak_list, xc=0, yc=0, r_max=52
)
assert (peak_filtered_list1 == np.array([[50, 0], [0, 30]])).all()
peak_filtered_list2 = ct._filter_peak_list_radius(
peak_list, xc=0, yc=0, r_max=81
)
assert len(peak_filtered_list2) == 2
peak_filtered_list3 = ct._filter_peak_list_radius(
peak_list, xc=0, yc=0, r_max=11
)
assert len(peak_filtered_list3) == 0
def test_wrong_input_no_r_min_or_r_max(self):
peak_list = np.array([[50, 10], [50, 50]])
with pytest.raises(ValueError):
ct._filter_peak_list_radius(peak_list, xc=10, yc=50)
