From a652a9b3e9442c8d3c77588b8dbb2ffa28ae655d Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Mon, 8 Apr 2024 13:29:03 -0700 Subject: [PATCH] test(test_models.py): combine add + delete model test --- tests/test_models.py | 21 ++++----------------- 1 file changed, 4 insertions(+), 17 deletions(-) diff --git a/tests/test_models.py b/tests/test_models.py index 21f41910c1..eb0cbbe5fd 100644 --- a/tests/test_models.py +++ b/tests/test_models.py @@ -127,20 +127,6 @@ async def chat_completion(session, key): raise Exception(f"Request did not return a 200 status code: {status}") -@pytest.mark.asyncio -async def test_add_models(): - """ - Add model - Call new model - """ - async with aiohttp.ClientSession() as session: - key_gen = await generate_key(session=session) - key = key_gen["key"] - await add_models(session=session) - await asyncio.sleep(60) - await chat_completion(session=session, key=key) - - @pytest.mark.asyncio async def test_get_models(): """ @@ -178,14 +164,15 @@ async def delete_model(session, model_id="123"): @pytest.mark.asyncio -async def test_delete_models(): +async def test_add_and_delete_models(): """ - Get models user has access to + Add model + Call new model """ - model_id = "12345" async with aiohttp.ClientSession() as session: key_gen = await generate_key(session=session) key = key_gen["key"] + model_id = "1234" await add_models(session=session, model_id=model_id) await asyncio.sleep(60) await chat_completion(session=session, key=key)