Add title field to ScoreRanges

Author: bencap

src/mavedb/view_models/score_range.py

@@ -87,6 +87,7 @@ class ScoreRange(SavedScoreRange):
 
 
 class ScoreRangesBase(BaseModel):
+    title: str = "Score Ranges"
     ranges: Sequence[ScoreRangeBase]
     research_use_only: bool = False
     source: Optional[Sequence[PublicationIdentifierBase]] = None
@@ -194,6 +195,7 @@ class InvestigatorScoreRange(ScoreRange, SavedInvestigatorScoreRange):
 
 
 class InvestigatorScoreRangesBase(ScoreRangesBase):
+    title: str = "Investigator-provided functional classes"
     baseline_score: Optional[float] = None
     baseline_score_description: Optional[str] = None
     ranges: Sequence[InvestigatorScoreRangeBase]
@@ -313,6 +315,7 @@ class ZeibergCalibrationParameterSet(BaseModel):
 
 
 class ZeibergCalibrationScoreRangesBase(ScoreRangesBase):
+    title: str = "Zeiberg calibration"
     prior_probability_pathogenicity: Optional[float] = None
     parameter_sets: list[ZeibergCalibrationParameterSet] = []
     ranges: Sequence[ZeibergCalibrationScoreRangeBase]

tests/helpers/constants.py

@@ -1601,6 +1601,7 @@
         TEST_SCORE_SET_ABNORMAL_RANGE,
     ],
     "research_use_only": False,
+    "title": "Test Base Ranges",
     "source": None,
 }
 
@@ -1612,6 +1613,7 @@
         TEST_SCORE_SET_ABNORMAL_RANGE,
     ],
     "research_use_only": False,
+    "title": "Test Base Ranges with Source",
     "source": [{"identifier": TEST_PUBMED_IDENTIFIER, "db_name": "PubMed"}],
 }
 
@@ -1663,6 +1665,7 @@
         TEST_INVESTIGATOR_PROVIDED_SCORE_SET_NOT_SPECIFIED_RANGE,
     ],
     "research_use_only": False,
+    "title": "Test Investigator-provided Functional Ranges",
     "odds_path_source": [{"identifier": TEST_PUBMED_IDENTIFIER, "db_name": "PubMed"}],
     "source": None,
 }
@@ -1677,6 +1680,7 @@
         TEST_SAVED_INVESTIGATOR_PROVIDED_SCORE_SET_NOT_SPECIFIED_RANGE,
     ],
     "researchUseOnly": False,
+    "title": "Test Investigator-provided Functional Ranges",
     "oddsPathSource": [{"identifier": TEST_PUBMED_IDENTIFIER, "dbName": "PubMed"}],
     "source": None,
 }
@@ -1876,6 +1880,7 @@
         TEST_ZEIBERG_CALIBRATION_SCORE_SET_PS3_VERY_STRONG_RANGE,
     ],
     "research_use_only": True,
+    "title": "Test Zeiberg Calibration",
     "parameter_sets": TEST_ZEIBERG_CALIBRATION_PARAMETER_SETS,
     "prior_probability_pathogenicity": 0.20,
     "source": None,
@@ -1895,6 +1900,7 @@
         TEST_ZEIBERG_CALIBRATION_SAVED_SCORE_SET_PS3_VERY_STRONG_RANGE,
     ],
     "researchUseOnly": True,
+    "title": "Test Zeiberg Calibration",
     "parameterSets": TEST_SAVED_ZEIBERG_CALIBRATION_PARAMETER_SETS,
     "priorProbabilityPathogenicity": 0.20,
 }

tests/view_models/test_score_range.py

@@ -361,6 +361,7 @@ def test_score_ranges_investigator_valid_range(ScoreRangesModel, score_ranges_da
     assert score_ranges.ranges is not None, "Ranges should not be None"
     assert score_ranges.baseline_score == TEST_BASELINE_SCORE, "Baseline score should match"
     assert score_ranges.research_use_only is False, "Research use only should be False for invesitigator provided"
+    assert score_ranges.title == score_ranges_data.get("title", None), "Title should match"
     assert matched_odds_source == score_ranges_data.get("odds_path_source", None), "Odds path source should match"
     assert matched_source == score_ranges_data.get("source", None), "Source should match"
 
@@ -386,6 +387,7 @@ def test_score_ranges_zeiberg_calibration_valid_range(ScoreRangesModel, score_ra
     ), "Prior probability pathogenicity should match"
     assert score_ranges.parameter_sets is not None, "Parameter sets should not be None"
     assert score_ranges.research_use_only is True, "Research use only should be True for zeiberg calibration"
+    assert score_ranges.title == score_ranges_data.get("title", None), "Title should match"
     assert matched_source == score_ranges_data.get("source", None), "Source should match"