Fix tsv tests

compomics · Aug 14, 2024 · dd9aac0 · dd9aac0
1 parent f6e4086
commit dd9aac0
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/tests/test_io/test_tsv.py b/tests/test_io/test_tsv.py
@@ -2,8 +2,9 @@
 
 import pytest
 
-from psm_utils.io.exceptions import PSMUtilsIOException # noqa: F401
-from psm_utils.io.tsv import TSVReader, TSVWriter # noqa: F401
+from psm_utils.io.exceptions import PSMUtilsIOException
+from psm_utils.io.tsv import TSVReader
+from psm_utils.peptidoform import Peptidoform
 
 test_cases = [
  (
@@ -37,7 +38,7 @@ def test__parse_entry(self):
  def test_iter(self):
  reader = TSVReader("tests/test_data/test.tsv")
  for psm in reader:
- assert psm.peptidoform == "ACDEK/2"
+ assert psm.peptidoform == Peptidoform("ACDEK/2")
  assert psm.spectrum_id == "peptide1"
  assert psm.provenance_data == {}
  assert psm.metadata == {}