From 4df95afbbbdd220700dee683acd716128f8e03af Mon Sep 17 00:00:00 2001 From: Jean Connelly Date: Thu, 12 Sep 2024 13:47:17 -0400 Subject: [PATCH] Handle bad stars from supplement in one test --- agasc/tests/test_agasc_2.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/agasc/tests/test_agasc_2.py b/agasc/tests/test_agasc_2.py index 93263c4..5237f2b 100644 --- a/agasc/tests/test_agasc_2.py +++ b/agasc/tests/test_agasc_2.py @@ -427,10 +427,14 @@ def test_supplement_get_agasc_cone(): ) ok = stars2["MAG_CATID"] == agasc.MAG_CATID_SUPPLEMENT - change_names = ["MAG_CATID", "COLOR1", "MAG_ACA", "MAG_ACA_ERR"] + change_names = ["MAG_CATID", "COLOR1", "MAG_ACA", "MAG_ACA_ERR", "CLASS"] for name in set(stars1.colnames) - set(change_names): assert np.all(stars1[name] == stars2[name]) + # For CLASS the only ones that are different should have the "bad" class 100 + class_nok = stars2["CLASS"] != stars1["CLASS"] + assert np.all(stars2["CLASS"][class_nok] == 100) + assert not np.any(stars1["MAG_CATID"] == agasc.MAG_CATID_SUPPLEMENT) # At least 35 stars in this field observed