add more integration tests

MannLabs · Nov 25, 2024 · 3e745b3 · 3e745b3
1 parent 673660b
commit 3e745b3
Show file tree

Hide file tree

Showing 4 changed files with 15 additions and 1 deletion.
diff --git a/alphabase/constants/const_files/psm_reader.yaml b/alphabase/constants/const_files/psm_reader.yaml
@@ -6,7 +6,7 @@ alphadia:
     'raw_name': 'run'
     'sequence': 'sequence'
     'charge': 'charge'
-    'rt': 'rt'
+    'rt': 'rt_observed'
     'rt_start': 'rt_start'
     'rt_stop': 'rt_stop'
     'ccs': 'ccs'

diff --git a/tests/integration/reference_data/reference_alphadia.parquet b/tests/integration/reference_data/reference_alphadia.parquet
diff --git a/tests/integration/reference_data/reference_alphadia_1.8.1_tsv.parquet b/tests/integration/reference_data/reference_alphadia_1.8.1_tsv.parquet
diff --git a/tests/integration/test_psm_readers.py b/tests/integration/test_psm_readers.py
@@ -441,6 +441,20 @@ def test_alphadia_reader() -> None:
     _assert_reference_df_equal(reader.psm_df, "alphadia", loose_check=True)
 
 
+def test_alphadia_reader_181() -> None:
+    """Test the AlphaDIA reader with extended input from v1.8.1."""
+
+    url = "https://datashare.biochem.mpg.de/s/Hk41INtwBvBl0kP/download?files=alphadia_1.8.1_report_head.tsv"
+    file_path = DataShareDownloader(
+        url=url, output_dir=current_file_directory + "/input_data/"
+    ).download()
+
+    reader = AlphaDiaReaderTsv()
+    reader.import_file(file_path)
+
+    _assert_reference_df_equal(reader.psm_df, "alphadia_1.8.1_tsv", loose_check=True)
+
+
 def test_diann_speclib_reader() -> None:
     """Test the Diann speclib reader."""
     # this is the head of  "https://datashare.biochem.mpg.de/s/DF12ObSdZnBnqUV" ("diann_speclib.tsv")