fix: Upgrade pytest-asyncio (#7746)

This commit is contained in:
Michael Genson
2026-06-09 16:09:35 -05:00
committed by GitHub
parent 040ec56c18
commit ac0bb4fb2c
4 changed files with 15 additions and 21 deletions

View File

@@ -101,9 +101,8 @@ def test_brute_parser(
comment: str,
):
with session_context() as session:
loop = asyncio.get_event_loop()
parser = get_parser(RegisteredParser.brute, unique_local_group_id, session, get_locale_provider())
parsed = loop.run_until_complete(parser.parse_one(input))
parsed = asyncio.run(parser.parse_one(input))
ing = parsed.ingredient
if ing.quantity:
@@ -145,15 +144,8 @@ def test_brute_parser_confidence(
input_str = f"1 {unit} {food}"
with session_context() as session:
original_loop = asyncio.get_event_loop()
try:
loop = asyncio.new_event_loop()
asyncio.set_event_loop(loop)
parser = get_parser(RegisteredParser.brute, unique_local_group_id, session, get_locale_provider())
parsed = loop.run_until_complete(parser.parse_one(input_str))
finally:
loop.close()
asyncio.set_event_loop(original_loop)
parser = get_parser(RegisteredParser.brute, unique_local_group_id, session, get_locale_provider())
parsed = asyncio.run(parser.parse_one(input_str))
conf = parsed.confidence

View File

@@ -50,9 +50,8 @@ def normalize(val: str) -> str:
)
def test_nlp_parser(unique_local_group_id: UUID4, test_ingredient: TestIngredient):
with session_context() as session:
loop = asyncio.get_event_loop()
parser = get_parser(RegisteredParser.nlp, unique_local_group_id, session, get_locale_provider())
parsed = loop.run_until_complete(parser.parse_one(test_ingredient.input))
parsed = asyncio.run(parser.parse_one(test_ingredient.input))
ing = parsed.ingredient
assert ing.quantity == pytest.approx(test_ingredient.quantity)

View File

@@ -56,11 +56,10 @@ def test_openai_parser(
monkeypatch.setattr(OpenAIService, "__init__", mock_openai_init)
with session_context() as session:
loop = asyncio.get_event_loop()
parser = get_parser(RegisteredParser.openai, unique_local_group_id, session, get_locale_provider())
inputs = [random_string() for _ in range(ingredient_count)]
parsed = loop.run_until_complete(parser.parse(inputs))
parsed = asyncio.run(parser.parse(inputs))
# since OpenAI is mocked, we don't need to validate the data, we just need to make sure parsing works
# and that it preserves order
@@ -109,10 +108,10 @@ def test_openai_parser_sanitize_output(
monkeypatch.setattr(OpenAIService, "__init__", mock_openai_init)
with session_context() as session:
loop = asyncio.get_event_loop()
parser = get_parser(RegisteredParser.openai, unique_local_group_id, session, get_locale_provider())
parsed = loop.run_until_complete(parser.parse([""]))
parsed = asyncio.run(parser.parse([""]))
assert len(parsed) == 1
parsed_ing = cast(ParsedIngredient, parsed[0])
assert parsed_ing.ingredient.food