diff --git a/tests/protzilla/data_analysis/test_filter_peptites_of_protein.py b/tests/protzilla/data_analysis/test_filter_peptites_of_protein.py index b191f335..23d739a2 100644 --- a/tests/protzilla/data_analysis/test_filter_peptites_of_protein.py +++ b/tests/protzilla/data_analysis/test_filter_peptites_of_protein.py @@ -1,10 +1,10 @@ import pytest -from protzilla.data_analysis.ptm_analysis import filter_peptides_of_protein +from protzilla.data_analysis.ptm_analysis import select_peptides_of_protein def test_filter_peptides_of_protein(peptides_df): - filtered_peptides_df = filter_peptides_of_protein(peptides_df, ["Protein2"])["peptide_df"] + filtered_peptides_df = select_peptides_of_protein(peptides_df, ["Protein2"])["peptide_df"] assert len(filtered_peptides_df) == 6 assert filtered_peptides_df["Sequence"].tolist() == [ diff --git a/tests/protzilla/data_analysis/test_peptide_analysis.py b/tests/protzilla/data_analysis/test_peptide_analysis.py index 864e9f17..8e87de88 100644 --- a/tests/protzilla/data_analysis/test_peptide_analysis.py +++ b/tests/protzilla/data_analysis/test_peptide_analysis.py @@ -1,14 +1,14 @@ import pytest -from protzilla.data_analysis.ptm_analysis import filter_peptides_of_protein, ptms_per_sample, \ +from protzilla.data_analysis.ptm_analysis import select_peptides_of_protein, ptms_per_sample, \ ptms_per_protein_and_sample @pytest.mark.parametrize("df_num", [0, 1]) -def test_filter_peptides_of_protein(peptides_df, evidence_peptide_df, df_num): +def test_select_peptides_of_protein(peptides_df, evidence_peptide_df, df_num): peptide_df = [peptides_df, evidence_peptide_df][df_num] - filtered_peptides_df = filter_peptides_of_protein(peptide_df, ["Protein2"])["peptide_df"] + filtered_peptides_df = select_peptides_of_protein(peptide_df, ["Protein2"])["peptide_df"] assert len(filtered_peptides_df) == 6 assert filtered_peptides_df["Sequence"].tolist() == [