From 1c9d91a29b261c795d4b7186e5352bebf1be2e1e Mon Sep 17 00:00:00 2001 From: Andy Kee Date: Fri, 1 Dec 2023 17:47:50 -0800 Subject: [PATCH] Update tests to match new collect_charge_bayer interface --- tests/test_detector.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/test_detector.py b/tests/test_detector.py index e24716a..923baf8 100644 --- a/tests/test_detector.py +++ b/tests/test_detector.py @@ -31,7 +31,7 @@ def test_collect_charge_bayer_even(): qe_red = np.random.uniform(size=5) qe_green = np.random.uniform(size=5) qe_blue = np.random.uniform(size=5) - bayer_pattern = [['R', 'G'], ['G', 'B']] + bayer_pattern = 'RGGB' out = lentil.detector.collect_charge_bayer(img, np.ones(5), qe_red, qe_green, qe_blue, bayer_pattern) @@ -48,7 +48,7 @@ def collect_charge_bayer_odd(): qe_red = np.random.uniform(size=5) qe_green = np.random.uniform(size=5) qe_blue = np.random.uniform(size=5) - bayer_pattern = [['R', 'G'], ['G', 'B']] + bayer_pattern = 'RGGB' out = lentil.detector.collect_charge_bayer(img, np.ones(5), qe_red, qe_green, qe_blue, bayer_pattern) @@ -70,7 +70,7 @@ def test_collect_charge_bayer_oversample(): qe_red = np.random.uniform(size=5) qe_green = np.random.uniform(size=5) qe_blue = np.random.uniform(size=5) - bayer_pattern = [['R', 'G'], ['G', 'B']] + bayer_pattern = 'RGGB' out = lentil.detector.collect_charge_bayer(img, np.ones(5), qe_red, qe_green, qe_blue, bayer_pattern, oversample=2)