From 1d210971272ab829c55c6f3e8f7309efdbc8651d Mon Sep 17 00:00:00 2001 From: Karthik Rishinarada Date: Mon, 13 Jan 2025 22:58:25 +0530 Subject: [PATCH 1/2] add test to line info --- .mailmap | 2 ++ tardis/visualization/widgets/tests/test_line_info.py | 7 +++++++ 2 files changed, 9 insertions(+) diff --git a/.mailmap b/.mailmap index b238be626b3..8cc367cbda4 100644 --- a/.mailmap +++ b/.mailmap @@ -119,6 +119,8 @@ Josh Shields Karan Desai Karan Desai karandesai-96 +Karthik Rishinarada + Kaushik Varanasi Kaushik Varanasi kaushik94 diff --git a/tardis/visualization/widgets/tests/test_line_info.py b/tardis/visualization/widgets/tests/test_line_info.py index 87686c3af29..bfc670ef29f 100644 --- a/tardis/visualization/widgets/tests/test_line_info.py +++ b/tardis/visualization/widgets/tests/test_line_info.py @@ -53,6 +53,13 @@ def test_get_species_interactions( ) assert species_interactions_df.shape == (expected_df_length, 1) + def test_get_middle_half_edges(self, wavelength_range, filter_mode): + arr = np.array([0, 1, 2, 3, 4]) + res = LineInfoWidget.get_middle_half_edges(arr) + arr = np.sort(arr) + expected_res = [(arr[-1] - arr[0]) / 4 + arr[1], (arr[-1] - arr[0]) * 3 / 4 + arr[1]] + assert res == expected_res + @pytest.fixture def allowed_species(self, line_info_widget, wavelength_range, filter_mode): """ From cd3b64de6fc36867d97ec7e776a6f4ff1519fd54 Mon Sep 17 00:00:00 2001 From: Karthik Rishinarada Date: Tue, 28 Jan 2025 10:02:14 +0530 Subject: [PATCH 2/2] using existing line_info_widget --- tardis/visualization/widgets/tests/test_line_info.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tardis/visualization/widgets/tests/test_line_info.py b/tardis/visualization/widgets/tests/test_line_info.py index bfc670ef29f..502de850b85 100644 --- a/tardis/visualization/widgets/tests/test_line_info.py +++ b/tardis/visualization/widgets/tests/test_line_info.py @@ -53,9 +53,9 @@ def test_get_species_interactions( ) assert species_interactions_df.shape == (expected_df_length, 1) - def test_get_middle_half_edges(self, wavelength_range, filter_mode): + def test_get_middle_half_edges(self, line_info_widget, wavelength_range, filter_mode): arr = np.array([0, 1, 2, 3, 4]) - res = LineInfoWidget.get_middle_half_edges(arr) + res = line_info_widget.get_middle_half_edges(arr) arr = np.sort(arr) expected_res = [(arr[-1] - arr[0]) / 4 + arr[1], (arr[-1] - arr[0]) * 3 / 4 + arr[1]] assert res == expected_res