Skip to content

Commit 9284f1f

Browse files
changes tests
1 parent 65549cb commit 9284f1f

File tree

1 file changed

+9
-11
lines changed

1 file changed

+9
-11
lines changed

tests/test_mumble.py

Lines changed: 9 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)