@@ -162,13 +162,12 @@ def test_add_modified_psms(self, setup_psmhandler):
162162
163163 def test_tool_combination_length_1 (self , setup_psm ):
164164
165- psm_handler = PSMHandler (combination_length = 1 , exclude_mutations = False )
165+ psm_handler = PSMHandler (
166+ combination_length = 1 , exclude_mutations = False , all_unimod_modifications = True
167+ )
166168
167169 # retrigger get_unimod_database
168- cache_file = psm_handler .modification_handler .cache ._get_cache_file_path ()
169- psm_handler .modification_handler .cache ._load_or_generate_data (
170- cache_file , force_reload = True
171- )
170+ psm_handler .modification_handler .cache .load_cache (force_reload = True )
172171 psm = setup_psm
173172
174173 result_psm_list = psm_handler .get_modified_peptidoforms_list (psm )
@@ -186,13 +185,10 @@ def test_tool_combination_length_1(self, setup_psm):
186185
187186 def test_tool_combination_length_2 (self , setup_psm ):
188187
189- psm_handler = PSMHandler (combination_length = 2 )
188+ psm_handler = PSMHandler (combination_length = 2 , all_unimod_modifications = True )
190189
191190 # retrigger get_unimod_database
192- cache_file = psm_handler .modification_handler .cache ._get_cache_file_path ()
193- psm_handler .modification_handler .cache ._load_or_generate_data (
194- cache_file , force_reload = True
195- )
191+ psm_handler .modification_handler .cache ._load_or_generate_data (force_reload = True )
196192
197193 psm = setup_psm
198194 result_psm_list = psm_handler .get_modified_peptidoforms_list (psm )
@@ -228,7 +224,7 @@ def test_tool_combination_length_2(self, setup_psm):
228224
229225 def test_tool_include_original_psm (self , setup_psm ):
230226 # psm_handler = setup_psmhandler[0]
231- psm_handler = PSMHandler (combination_length = 1 )
227+ psm_handler = PSMHandler (combination_length = 1 , all_unimod_modifications = True )
232228
233229 psm = setup_psm
234230
@@ -782,6 +778,7 @@ def test_single_combined_modifcations(self):
782778 mass_error = 0.02 ,
783779 exclude_mutations = True ,
784780 combination_length = 1 ,
781+ all_unimod_modifications = True ,
785782 )
786783
787784 mapped_psms = psm_handler .add_modified_psms (
@@ -825,6 +822,7 @@ def test_double_combined_modifcations(self):
825822 mass_error = 0.02 ,
826823 exclude_mutations = True ,
827824 combination_length = 2 ,
825+ all_unimod_modifications = True ,
828826 )
829827
830828 mapped_psms = psm_handler .add_modified_psms (
0 commit comments