From a9ad5f09bebfff0df4c5ba7543d211915b75354d Mon Sep 17 00:00:00 2001 From: Konstantin Malanchev Date: Mon, 7 Oct 2024 14:03:17 -0400 Subject: [PATCH] Update tests/tdastro/sources/test_snia.py Co-authored-by: mi-dai --- tests/tdastro/sources/test_snia.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/tdastro/sources/test_snia.py b/tests/tdastro/sources/test_snia.py index 8a8dbca2..6a697980 100644 --- a/tests/tdastro/sources/test_snia.py +++ b/tests/tdastro/sources/test_snia.py @@ -35,7 +35,7 @@ def draw_single_random_sn( # Update obs_index to only include observations within SN lifespan phase_obs = opsim[opsim.colmap["time"]][obs_index] - t0 - obs_index = obs_index[(phase_obs > -20) & (phase_obs < 50)] + obs_index = obs_index[(phase_obs > -20 * (1.0 + z)) & (phase_obs < 50 * (1.0 + z))] times = opsim[opsim.colmap["time"]][obs_index].to_numpy() filters = opsim[opsim.colmap["filter"]][obs_index].to_numpy(str)