From d350cee32f5ea9b3691ff5325f51a8eca5a53887 Mon Sep 17 00:00:00 2001 From: Francisco Aranda Date: Wed, 26 Jun 2024 16:06:02 +0200 Subject: [PATCH] chore: Simplify test setup --- argilla/tests/unit/test_interface.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/argilla/tests/unit/test_interface.py b/argilla/tests/unit/test_interface.py index 85c72faa63b..dbbae5acca2 100644 --- a/argilla/tests/unit/test_interface.py +++ b/argilla/tests/unit/test_interface.py @@ -29,8 +29,8 @@ def test_default_client(self): assert client.api_key == "admin.apikey" def test_multiple_clients(self): - with mock.patch("argilla.client._api.APIClient.http_client"): - local_client = rg.Argilla(api_url="http://localhost:6900", api_key="admin.apikey") - remote_client = rg.Argilla(api_url="http://argilla.production.net", api_key="admin.apikey") - assert local_client.api_url == "http://localhost:6900" - assert remote_client.api_url == "http://argilla.production.net" + local_client = rg.Argilla(api_url="http://localhost:6900", api_key="admin.apikey") + remote_client = rg.Argilla(api_url="http://argilla.production.net", api_key="admin.apikey") + + assert local_client.api_url == "http://localhost:6900" + assert remote_client.api_url == "http://argilla.production.net"