fixing test_metrics_fakecomp
@@ -3,7 +3,7 @@ Tests for metrics/fakecomp.py
"""
-__author__ = "Ralf Farkas"
+__author__ = "Ralf Farkas, Carlos Alberto Gomez Gonzalez"
from .helpers import aarc, np, param, parametrize, fixture, filterwarnings
from vip_hci.metrics.fakecomp import cube_inject_companions, normalize_psf
@@ -18,10 +18,10 @@ def dataset(request):
"""
if request.param == "3D":
- cube = np.zeros((3, 5, 5))
+ cube = np.zeros((3, 25, 25))
psf = np.ones((1, 1))
elif request.param == "4D":
- cube = np.zeros((2, 3, 5, 5)) # lambda, frames, width, height
+ cube = np.zeros((2, 3, 25, 25)) # lambda, frames, width, height
psf = np.ones((2, 1, 1))
angles = np.array([0, 90, 180])
@@ -44,12 +44,11 @@ def test_cube_inject_companions(dataset, branches, dists):
Expected positions.
"""
if branches == 1 and dists == 2:
- return [(2, 4)]
+ return [(12, 14)]
elif branches == 2 and dists == 2:
- return [(2, 4), (2, 0)]
+ return [(12, 14), (12, 10)]
elif branches == 2 and dists == [1, 2]:
- return [(2, 3), (2, 4), (2, 1), (2, 0)]
-
+ return [(12, 13), (12, 14), (12, 11), (12, 10)]
else:
raise ValueError("no expected result defined")
