From 112e71bb00e4066d245798a4cc278755c6aa47ad Mon Sep 17 00:00:00 2001 From: Matthew The Date: Thu, 2 Jul 2026 20:56:38 +0200 Subject: [PATCH 1/2] Add unit test for the new Probability column functionality --- tests/unit_tests/parsers/test_fragpipe.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/tests/unit_tests/parsers/test_fragpipe.py b/tests/unit_tests/parsers/test_fragpipe.py index 59e5f72..19cf357 100644 --- a/tests/unit_tests/parsers/test_fragpipe.py +++ b/tests/unit_tests/parsers/test_fragpipe.py @@ -1,5 +1,3 @@ -# Generation assisted by CodiumAI - from picked_group_fdr.parsers.fragpipe import parse_fragpipe_psm_file from picked_group_fdr.scoring_strategy import ProteinScoringStrategy @@ -10,7 +8,10 @@ class TestParseFragpipePsmFile: # correctly parses a well-formed psm.tsv file with all required columns - def test_correctly_parses_well_formed_psm_tsv(self, mocker): + @pytest.mark.parametrize( + "probability_column", ["PeptideProphet Probability", "Probability"] + ) + def test_correctly_parses_well_formed_psm_tsv(self, mocker, probability_column): # Mock the csv reader mocker.patch( "picked_group_fdr.parsers.tsv.get_tsv_reader", @@ -22,7 +23,7 @@ def test_correctly_parses_well_formed_psm_tsv(self, mocker): "Peptide", "Modified Peptide", "SpectralSim", - "PeptideProphet Probability", + probability_column, "Protein", "Mapped Proteins", ], From da71b42dfa9bba07a8c6ba789eea15554e9c3c0b Mon Sep 17 00:00:00 2001 From: Matthew The Date: Thu, 2 Jul 2026 21:00:51 +0200 Subject: [PATCH 2/2] Ready for v0.9.1