fix: add list_runs to tests

Author: Michał Sośnicki

tests/integration/neptune2/test_neptune2.py

@@ -1,3 +1,4 @@
+import re
 from typing import Generator
 import json
 import pyarrow as pa
@@ -7,6 +8,16 @@
 from .data import TEST_DATA
 
 
+def test_neptune2_list_runs(api_token, project, test_runs):
+    exporter = Neptune2Exporter(api_token=api_token)
+
+    runs = exporter.list_runs(
+        project_id=project, runs="|".join(re.escape(run_id) for run_id in test_runs)
+    )
+
+    assert len(runs) == len(test_runs)
+
+
 def test_neptune2_download_parameters_empty(api_token, project, test_runs):
     exporter = Neptune2Exporter(api_token=api_token)
 

tests/integration/neptune3/test_neptune3.py

@@ -1,10 +1,21 @@
+import re
 from typing import Generator
 import pyarrow as pa
 from neptune_exporter import model
 from neptune_exporter.exporters.neptune3 import Neptune3Exporter
 from .data import TEST_DATA
 
 
+def test_neptune3_list_runs(api_token, project, test_runs):
+    exporter = Neptune3Exporter(api_token=api_token)
+
+    runs = exporter.list_runs(
+        project_id=project, runs="|".join(re.escape(run_id) for run_id in test_runs)
+    )
+
+    assert len(runs) == len(test_runs)
+
+
 def test_neptune3_download_parameters_empty(api_token, project, test_runs):
     exporter = Neptune3Exporter(api_token=api_token)