Sleep longer, allow extraAttributes

multimeric · Aug 14, 2024 · b7f1c31 · b7f1c31
1 parent d6e8469
commit b7f1c31
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/test/test_dataset/test_request.py b/test/test_dataset/test_request.py
@@ -88,6 +88,7 @@ def test_main_example():
     ).each_record():
         assert isinstance(record, dict)
         assert set(record.keys()) == {
+            "extraAttributes",
             "primaryAccession",
             "genes",
             "sequence",

diff --git a/test/test_id_mapping/test_id_mapping.py b/test/test_id_mapping/test_id_mapping.py
@@ -126,7 +126,7 @@ def test_gene_names():
     )
     # At this point it might be already finished
     assert request.get_status() in {"RUNNING", "FINISHED"}
-    sleep(1)
+    sleep(5)
     # At this point it should definitely be finished
     assert request.get_status() == "FINISHED"
     for result in request.each_result():