Merge branch 'pp/extra_search_engines' into pp/se_api

NREL · Feb 6, 2025 · 2f01161 · 2f01161
2 parents 60bbf13 + e44eec8
commit 2f01161
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 6 deletions.
diff --git a/tests/web/search/test_web_search.py b/tests/web/search/test_web_search.py
@@ -25,9 +25,9 @@
 
 
 @flaky(max_runs=3, min_passes=1)
-@pytest.mark.parametrize("queries",[['1. "Python Programming Language"'],
-                                    ['1. "Python Programming Language"',
-                                     "Python"],])
+@pytest.mark.parametrize("queries", [['1. "Python Programming Language"'],
+                                     ['1. "Python Programming Language"',
+                                      "Python"],])
 @pytest.mark.parametrize("se", SE_TO_TEST)
 @pytest.mark.asyncio
 async def test_basic_search_query(queries, se):
@@ -68,9 +68,9 @@ async def _tps(obj, page, search_query):
     assert not out[1]
 
 
-@pytest.mark.parametrize("queries",[['1. "Python Programming Language"'],
-                                    ['1. "Python Programming Language"',
-                                     "Python"],])
+@pytest.mark.parametrize("queries", [['1. "Python Programming Language"'],
+                                     ['1. "Python Programming Language"',
+                                      "Python"],])
 @pytest.mark.parametrize("num_results_multiplier", [0, 1, 5])
 @pytest.mark.asyncio
 async def test_search_query_num_results(queries, num_results_multiplier):

diff --git a/tests/web/test_osti.py b/tests/web/test_osti.py
@@ -4,6 +4,9 @@
 """
 import os
 import tempfile
+
+from flaky import flaky
+
 from elm import OstiList
 
 
@@ -27,6 +30,7 @@ def test_osti_from_url():
     assert len(docs) == 12
 
 
+@flaky(max_runs=5, min_passes=1)
 def test_osti_from_oids():
     """Test osti list, make sure we can find specific oids from storage futures
     study"""