From 9cd925324bb3bff5f7d6d4b996cd6cc0859c42ff Mon Sep 17 00:00:00 2001 From: Alexander Long Date: Sat, 22 Feb 2025 14:37:10 -0700 Subject: [PATCH] adjusted test_full_roundtrip to handle list of radiusParameters --- tests/unit/pleiades/sammy/test_parfile.py | 15 +++++++++++---- 1 file changed, 11 insertions(+), 4 deletions(-) diff --git a/tests/unit/pleiades/sammy/test_parfile.py b/tests/unit/pleiades/sammy/test_parfile.py index f866eca..5c5de31 100644 --- a/tests/unit/pleiades/sammy/test_parfile.py +++ b/tests/unit/pleiades/sammy/test_parfile.py @@ -59,6 +59,8 @@ def test_parse_single_card(single_card_input): """Test parsing file with fudge factor and single broadening card.""" parfile = SammyParameterFile.from_string(single_card_input) + print(parfile) + # Check fudge factor assert parfile.fudge == pytest.approx(0.1) @@ -267,10 +269,15 @@ def test_full_roundtrip(self, temp_dir): print(loaded_parfile.radius) assert loaded_parfile.radius is not None assert orig_parfile.radius is not None - assert loaded_parfile.radius.parameters.effective_radius == pytest.approx(orig_parfile.radius.parameters.effective_radius) - assert loaded_parfile.radius.parameters.true_radius == pytest.approx(orig_parfile.radius.parameters.true_radius) - assert loaded_parfile.radius.parameters.spin_groups == orig_parfile.radius.parameters.spin_groups - assert loaded_parfile.radius.parameters.vary_effective == orig_parfile.radius.parameters.vary_effective + + # both original and loaded radius parameters should be length 1 + assert len(loaded_parfile.radius.parameters) == 1 + assert len(orig_parfile.radius.parameters) == 1 + + assert loaded_parfile.radius.parameters[0].effective_radius == pytest.approx(orig_parfile.radius.parameters[0].effective_radius) + assert loaded_parfile.radius.parameters[0].true_radius == pytest.approx(orig_parfile.radius.parameters[0].true_radius) + assert loaded_parfile.radius.parameters[0].spin_groups == orig_parfile.radius.parameters[0].spin_groups + assert loaded_parfile.radius.parameters[0].vary_effective == orig_parfile.radius.parameters[0].vary_effective # Compare data reduction parameters assert loaded_parfile.data_reduction is not None