Skip to content

Commit aaf15b1

Browse files
committed
Adjust lipidomics test to use new molecular formula calculations
1 parent 06e664a commit aaf15b1

1 file changed

Lines changed: 5 additions & 21 deletions

File tree

tests/test_wf_lipidomics.py

Lines changed: 5 additions & 21 deletions
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@
1313
from corems.mass_spectra.output.export import LipidomicsExport
1414
from corems.molecular_id.search.database_interfaces import MetabRefLCInterface
1515
from corems.molecular_id.factory.lipid_molecular_metadata import LipidMetadata
16-
from corems.molecular_id.search.molecularFormulaSearch import SearchMolecularFormulas
16+
from corems.molecular_id.search.molecularFormulaSearch import SearchMolecularFormulasLC
1717
from corems.encapsulation.factory.parameters import LCMSParameters, reset_lcms_parameters, reset_ms_parameters
1818

1919

@@ -82,7 +82,7 @@ def test_lipidomics_workflow():
8282
ms1_params.mass_spectrum.noise_min_mz, ms1_params.mass_spectrum.min_picking_mz = 0, 0
8383
ms1_params.mass_spectrum.noise_max_mz, ms1_params.mass_spectrum.max_picking_mz = np.inf, np.inf
8484
ms1_params.ms_peak.legacy_resolving_power = False
85-
ms1_params.molecular_search.url_database = "postgresql://coremsdb:coremsmolform@postgres:5432/molformula"
85+
ms1_params.molecular_search.url_database = ""
8686
ms1_params.molecular_search.usedAtoms = {
8787
'C': (10, 30),
8888
'H': (18, 200),
@@ -129,28 +129,12 @@ def test_lipidomics_workflow():
129129
assert len(myLCMSobj.mass_features) == 130
130130

131131
# Perform a molecular search on a few of the mass features
132-
mf_df = myLCMSobj.mass_features_to_df()
133-
unique_scans = mf_df.apex_scan.unique()
134-
i = 0
135-
for scan in unique_scans:
136-
if i > 1: # only search first 3 scans for testing
137-
break
138-
print("searching mz for scan: ", str(i), " of ", str(len(unique_scans)))
139-
# gather mass features for this scan
140-
mf_df_scan = mf_df[mf_df.apex_scan == scan]
141-
peaks_to_search = [
142-
myLCMSobj.mass_features[x].ms1_peak for x in mf_df_scan.index.tolist()
143-
]
144-
SearchMolecularFormulas(
145-
myLCMSobj._ms[scan],
146-
first_hit=False,
147-
find_isotopologues=True,
148-
).run_worker_ms_peaks(peaks_to_search)
149-
i += 1
132+
mol_form_search = SearchMolecularFormulasLC(myLCMSobj)
133+
mol_form_search.run_mass_feature_search()
150134

151135
# Check results of molecular search
152136
assert myLCMSobj.mass_features[0].ms1_peak[0].string == "C20 H30 O2"
153-
assert myLCMSobj.mass_features_ms1_annot_to_df().shape[0] == 130
137+
assert myLCMSobj.mass_features_ms1_annot_to_df().shape[0] == 257
154138
myLCMSobj.mass_features[0].mass_spectrum.to_dataframe()
155139

156140
# Add hcd ms2 data to lcms object, using the ms2 mass spectrum parameters

0 commit comments

Comments
 (0)