diff --git a/tests/test_models.py b/tests/test_models.py index 21f41910c..eb0cbbe5f 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)