Update test_spectrum_document.py

iomega · Aug 6, 2024 · 13033b5 · 13033b5
1 parent 1540f95
commit 13033b5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_spectrum_document.py b/tests/test_spectrum_document.py
@@ -41,7 +41,7 @@ def test_spectrum_document_init_default_with_losses():
     intensities = np.array([0, 0.01, 0.1, 1], dtype="float")
     metadata = dict(precursor_mz=100.0)
     spectrum_in = Spectrum(mz=mz, intensities=intensities, metadata=metadata)
-    spectrum_document = SpectrumDocument(spectrum)
+    spectrum_document = SpectrumDocument(spectrum_in)
 
     assert spectrum_document.n_decimals == 2, "Expected different default for n_decimals"
     assert len(spectrum_document) == 8
@@ -58,7 +58,7 @@ def test_spectrum_document_init_n_decimals_1():
     intensities = np.array([0, 0.01, 0.1, 1], dtype="float")
     metadata = dict(precursor_mz=100.0)
     spectrum_in = Spectrum(mz=mz, intensities=intensities, metadata=metadata)
-    spectrum_document = SpectrumDocument(spectrum, n_decimals=1)
+    spectrum_document = SpectrumDocument(spectrum_in, n_decimals=1)
 
     assert spectrum_document.n_decimals == 1
     assert len(spectrum_document) == 8