diff --git a/tests/test_cli.py b/tests/test_cli.py index 7c776f6..3b7b2e0 100644 --- a/tests/test_cli.py +++ b/tests/test_cli.py @@ -9,7 +9,16 @@ # local from alphamap.importing import read_file, \ - extract_rawfile_unique_values + extract_rawfile_unique_values, \ + import_spectronaut_data, \ + import_maxquant_data, \ + convert_ap_mq_mod, \ + import_alphapept_data, \ + convert_diann_mq_mod, \ + import_diann_data, \ + convert_fragpipe_mq_mod, \ + import_fragpipe_data, \ + import_data THIS_FOLDER = os.path.dirname(__file__) TEST_FOLDER = os.path.join( @@ -49,7 +58,218 @@ def test_extract_rawfile_unique_values(self, ): data_fragpipe_combined = os.path.join(TEST_FOLDER, 'combined_peptide.txt') assert ['Y731F1', 'Y731F2', 'wt1', 'wt2'] == extract_rawfile_unique_values(data_fragpipe_combined) + def test_import_spectronaut_data(self, ): + # test entire input test data + data = import_spectronaut_data(os.path.join(TEST_FOLDER, 'test_spectronaut_input.csv')) + #print(data.shape[0]) + assert data.shape[0] == 40 + data_t = import_spectronaut_data(os.path.join(TEST_FOLDER, 'test_spectronaut_input.tsv')) + #print(data_t.shape[0]) + pd.testing.assert_frame_equal(data, data_t) + test = pd.read_csv(os.path.join(TEST_FOLDER, 'test_spectronaut_imported.csv'), sep=',') + #print(test.shape[0]) + pd.testing.assert_frame_equal(data, test) + + # test single sample + data = import_spectronaut_data(os.path.join(TEST_FOLDER, 'test_spectronaut_input.csv'), + sample="raw_01") + #print(data.shape[0]) + assert data.shape[0] == 40 + data = import_spectronaut_data(os.path.join(TEST_FOLDER, 'test_spectronaut_input.csv'), + sample="raw_02") + #print(data.shape[0]) + assert data.shape[0] == 20 + + # test multiple samples + data = import_spectronaut_data(os.path.join(TEST_FOLDER, 'test_spectronaut_input.csv'), + sample=["raw_01","raw_02"]) + #print(data.shape[0]) + assert data.shape[0] == 40 + + def test_import_maxquant_data(self, ): + data = import_maxquant_data(os.path.join(TEST_FOLDER, 'test_maxquant_input.txt')) + test = pd.read_csv(os.path.join(TEST_FOLDER, 'test_maxquant_imported.csv'), sep=',') + pd.testing.assert_frame_equal(data, test) + + data_s = import_maxquant_data(os.path.join(TEST_FOLDER, 'test_maxquant_input.txt'), + sample = "raw_1") + assert data_s.shape[0] == 85 + + data_s = import_maxquant_data(os.path.join(TEST_FOLDER, 'test_maxquant_input.txt'), + sample = "raw_2") + assert data_s.shape[0] == 77 + + data_s = import_maxquant_data(os.path.join(TEST_FOLDER, 'test_maxquant_input.txt'), + sample = ["raw_1", "raw_2"]) + assert data_s.shape[0] == 136 + + def test_convert_ap_mq_mod(self, ): + # test oxidation + seq1 = 'HAEoxMVHTGLK' + assert "HAEM[Oxidation (M)]VHTGLK" == convert_ap_mq_mod(seq1) + seq2 = 'HAEoxPVHTGLK' + assert "HAEP[Oxidation (MP)]VHTGLK" == convert_ap_mq_mod(seq2) + # test acetylation + seq4 = 'aMDEPSPLAQPLELNQHSR' + assert "[Acetyl (N-term)]MDEPSPLAQPLELNQHSR" == convert_ap_mq_mod(seq4) + seq5 = 'MDEPSaKPLA' + assert "MDEPSK[Acetyl (K)]PLA" == convert_ap_mq_mod(seq5) + # test amidation + seq6 = 'MDEPSKPLamA' + assert "MDEPSKPLA[Amidated (C-term)]" == convert_ap_mq_mod(seq6) + # test deamidation + seq7 = 'MDEPSdeamNKPLA' + assert "MDEPSN[Deamidation (NQ)]KPLA" == convert_ap_mq_mod(seq7) + seq8 = 'MDEPSdeamQKPLA' + assert "MDEPSQ[Deamidation (NQ)]KPLA" == convert_ap_mq_mod(seq8) + # test phosporylation + seq9 = 'MDEPSpSKPLA' + assert "MDEPSS[Phospho (STY)]KPLA" == convert_ap_mq_mod(seq9) + # test pyro-Glu + seq10 = 'MDpgEPSNKPLA' + assert "MDE[Glu->pyro-Glu]PSNKPLA" == convert_ap_mq_mod(seq10) + seq11 = 'MDpgQPSNKPLA' + assert "MDQ[Gln->pyro-Glu]PSNKPLA" == convert_ap_mq_mod(seq11) + # test disylfide bonds + seq12 = 'cCVNTTLQIK' + assert "C[Cys-Cys]VNTTLQIK" == convert_ap_mq_mod(seq12) + seq1_several_dif_mods = 'AcCLDYPVTSVLPPASLoxMK' + assert "AC[Cys-Cys]LDYPVTSVLPPASLM[Oxidation (M)]K" == convert_ap_mq_mod(seq1_several_dif_mods) + seq2_several_same_mods = 'LFTToxMELoxMR' + assert "LFTTM[Oxidation (M)]ELM[Oxidation (M)]R" == convert_ap_mq_mod(seq2_several_same_mods) + seq3_several_same_mods = 'LFTToxMELoxMRoxM' + assert "LFTTM[Oxidation (M)]ELM[Oxidation (M)]RM[Oxidation (M)]" == convert_ap_mq_mod(seq3_several_same_mods) + seq4_several_same_mods = 'LFTTdeamNELdeamNR' + assert "LFTTN[Deamidation (NQ)]ELN[Deamidation (NQ)]R" == convert_ap_mq_mod(seq4_several_same_mods) + seq_no_mod = 'CVNTTLQIK' + assert "CVNTTLQIK" == convert_ap_mq_mod(seq_no_mod) + + def test_import_alphapept_data(self, ): + data = import_alphapept_data(os.path.join(TEST_FOLDER, 'test_alphapept_input.csv')) + assert data.shape == (4228, 3) + + data_s1 = import_alphapept_data(os.path.join(TEST_FOLDER, 'test_alphapept_input.csv'), + sample = "exp_1") + assert data_s1.shape[0] == 2127 + + data_s2 = import_alphapept_data(os.path.join(TEST_FOLDER, 'test_alphapept_input.csv'), + sample = "exp_2") + assert data_s2.shape[0] == 2101 + + data_s_both = import_alphapept_data(os.path.join(TEST_FOLDER, 'test_alphapept_input.csv'), + sample = ["exp_1", "exp_2"]) + assert data_s_both.shape[0] == 4228 + + def test_convert_diann_mq_mod(self, ): + seq1 = 'VSHGSSPSLLEALSSDFLAC(UniMod:4)K' + assert 'VSHGSSPSLLEALSSDFLAC[Carbamidomethyl (C)]K' == convert_diann_mq_mod(seq1) + seq2 = 'VSVINTVDTSHEDMIHDAQM(UniMod:35)DYYGTR' + assert 'VSVINTVDTSHEDMIHDAQM[Oxidation (M)]DYYGTR' == convert_diann_mq_mod(seq2) + seq3 = 'HAEMPVHTGLK(UniMod:2)' + assert 'HAEMPVHTGLK[Amidated (C-term)]' == convert_diann_mq_mod(seq3) + seq4 = 'HAEMPVHTGLKS(UniMod:23)A' + assert 'HAEMPVHTGLKS[Dehydrated (ST)]A' == convert_diann_mq_mod(seq4) + seq5 = 'HAEMPVHTGLKY(UniMod:23)A' + assert 'HAEMPVHTGLKY[Dehydrated (Y)]A' == convert_diann_mq_mod(seq5) + + seq1_several_dif_mods = '(UniMod:1)VSHGSSPSLLEALSSDFLAC(UniMod:4)K' + assert '[Acetyl (N-term)]VSHGSSPSLLEALSSDFLAC[Carbamidomethyl (C)]K' == convert_diann_mq_mod(seq1_several_dif_mods) + seq2_several_same_mods = 'CAALVATAEENLC(UniMod:4)C(UniMod:4)EELSSK' + assert 'CAALVATAEENLC[Carbamidomethyl (C)]C[Carbamidomethyl (C)]EELSSK' == convert_diann_mq_mod(seq2_several_same_mods) + seq_no_mod = 'CVNTTLQIK' + assert "CVNTTLQIK" == convert_diann_mq_mod(seq_no_mod) + + def test_import_diann_data(self, ): + data = import_diann_data(os.path.join(TEST_FOLDER, 'test_diann_input.tsv')) + assert data.shape == (44, 3) + + data_s1 = import_diann_data(os.path.join(TEST_FOLDER, 'test_diann_input.tsv'), + sample = "20201218_tims03_Evo03_PS_SA_HeLa_200ng_high_speed_21min_8cm_S2-B2_1_22648") + assert data_s1.shape[0] == 39 + + data_s_both = import_diann_data(os.path.join(TEST_FOLDER, 'test_diann_input.tsv'), + sample = ["20201218_tims03_Evo03_PS_SA_HeLa_200ng_high_speed_21min_8cm_S2-B2_1_22648", + "20201218_tims03_Evo03_PS_SA_HeLa_200ng_high_speed_21min_8cm_S2-A2_1_22636"]) + assert data_s_both.shape[0] == 42 + def test_convert_fragpipe_mq_mod(self, ): + seq1 = 'AAEREPPPLGDGKPTDFEDLEDGEDLFTSTVSTLE' + modif1 = 'N-term(42.0106)' + assert '[Acetyl (N-term)]AAEREPPPLGDGKPTDFEDLEDGEDLFTSTVSTLE' == convert_fragpipe_mq_mod(seq1, modif1) + seq2 = 'AAEVISDARENIQRFFGHGAEDSLADQAANEWGRSGKDPNHFRPAGLPEKY' + modif2 = 'C-term(-0.9840)' + assert 'AAEVISDARENIQRFFGHGAEDSLADQAANEWGRSGKDPNHFRPAGLPEKY[Amidated (C-term)]' == convert_fragpipe_mq_mod(seq2, modif2) + seq3 = 'QESQSEEIDCNDKDLFKA' + modif3 = '1Q(-17.0265)' + assert 'Q[Gln->pyro-Glu]ESQSEEIDCNDKDLFKA' == convert_fragpipe_mq_mod(seq3, modif3) + seq4 = 'EKPLLEKSHCIAEVENDEMPA' + modif4 = '1E(-18.0106)' + assert 'E[Glu->pyro-Glu]KPLLEKSHCIAEVENDEMPA' == convert_fragpipe_mq_mod(seq4, modif4) + seq5 = 'SKPLLEKSHCIAEVENDEMPA' + modif5 = '1S(-18.0106)' + assert 'S[Dehydrated (ST)]KPLLEKSHCIAEVENDEMPA' == convert_fragpipe_mq_mod(seq5, modif5) + seq6 = 'YKPLLEKSHCIAEVENDEMPA' + modif6 = '1Y(-18.0106)' + assert 'Y[Dehydrated (Y)]KPLLEKSHCIAEVENDEMPA' == convert_fragpipe_mq_mod(seq6, modif6) + seq1_several_dif_mods = 'AAAAECDVVMAATEPELLDDQEAK' + seq1_several_dif_mods_modifs = '10M(15.9949),6C(57.0215),N-term(42.0106)' + assert '[Acetyl (N-term)]AAAAEC[Carbamidomethyl (C)]DVVM[Oxidation (M)]AATEPELLDDQEAK' == convert_fragpipe_mq_mod( + seq1_several_dif_mods, seq1_several_dif_mods_modifs) + seq2_several_dif_mods = 'PGFSIADKKR' + seq2_several_dif_mods_modifs = '8K(114.0429),9K(114.0429)' + assert 'PGFSIADK[GlyGly (K)]K[GlyGly (K)]R' == convert_fragpipe_mq_mod( + seq2_several_dif_mods, seq2_several_dif_mods_modifs) + seq_no_mod = 'CVNTTLQIK' + seq_no_mod_modifs = '' + assert "CVNTTLQIK" == convert_fragpipe_mq_mod(seq_no_mod, seq_no_mod_modifs) + + def test_import_fragpipe_data(self, ): + data = import_fragpipe_data(os.path.join(TEST_FOLDER, 'test_fragpipe_input.tsv')) + assert data.shape == (50, 3) + + sample = 'Y731F1' + file = os.path.join(TEST_FOLDER, 'combined_peptide.txt') + assert import_fragpipe_data(file, sample).shape == (23, 3) + + two_samples = ['Y731F1', 'Y731F2'] + assert import_fragpipe_data(file, two_samples).shape == (26, 3) + + def test_import_data(self, ): + data_MQ = import_data(os.path.join(TEST_FOLDER, 'test_maxquant_input.txt'), verbose=False) + test = pd.read_csv(os.path.join(TEST_FOLDER, 'test_maxquant_imported.csv'), sep=',') + pd.testing.assert_frame_equal(data_MQ, test) + + data_S_csv = import_data(os.path.join(TEST_FOLDER, 'test_spectronaut_input.csv'), verbose=False) + data_S_tsv = import_data(os.path.join(TEST_FOLDER, 'test_spectronaut_input.tsv'), verbose=False) + pd.testing.assert_frame_equal(data_S_csv, data_S_tsv) + test = pd.read_csv(os.path.join(TEST_FOLDER, 'test_spectronaut_imported.csv'), sep=',') + pd.testing.assert_frame_equal(data_S_csv, test) + + data_S_sub = import_data(os.path.join(TEST_FOLDER, 'test_spectronaut_input.csv'), + sample = "raw_01", + verbose=False) + assert data_S_sub.shape[0] == 40 + + data_alphapept = import_data(os.path.join(TEST_FOLDER, 'test_alphapept_input.csv'), + sample = "exp_1", + verbose=False) + assert data_alphapept.shape[0] == 2127 + + data_diann = import_data(os.path.join(TEST_FOLDER, 'test_diann_input.tsv'), + sample = "20201218_tims03_Evo03_PS_SA_HeLa_200ng_high_speed_21min_8cm_S2-B2_1_22648", + verbose=False) + assert data_diann.shape[0] == 39 + + data_fragpipe = import_data(os.path.join(TEST_FOLDER, 'test_fragpipe_input.tsv'), + verbose=False) + assert data_fragpipe.shape[0] == 50 + + try: + out = import_data(os.path.join(TEST_FOLDER, 'test_uniprot_df.csv')) + except TypeError as e: + out = e + assert str(out) == "Input data format for "+os.path.join(TEST_FOLDER, 'test_uniprot_df.csv')+" not known." + if __name__ == "__main__": unittest.main()