wip

pmateusz · Dec 29, 2024 · 30e95ea · 30e95ea
1 parent cf59d4e
commit 30e95ea
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 9 deletions.
diff --git a/tests/client/aiohttp/test_array_query_params.py b/tests/client/aiohttp/test_array_query_params.py
@@ -14,7 +14,7 @@ async def test_handles_array_query_params(http_server: HTTPTestServer) -> None:
     # GIVEN
     http_server.handler = companies_filter_by_sector
 
-    @dataclass(slots=True)
+    @dataclass
     class Company:
         name: str
         sector: str
@@ -31,5 +31,7 @@ async def get_companies(
         companies = await client.get_companies(sectors=["Information Technology", "Financials"])
 
     # THEN
-    assert len(companies) > 0
-    assert all(company.sector in ["Information Technology", "Financials"] for company in companies)
+    assert companies == [
+        Company(name="Apple", sector="Information Technology"),
+        Company(name="Berkshire Hathaway", sector="Financials"),
+    ]
diff --git a/tests/client/httpx/test_array_query_params.py b/tests/client/httpx/test_array_query_params.py
@@ -12,7 +12,7 @@ def test_handles_array_query_params(http_server: HTTPTestServer) -> None:
     # GIVEN
     http_server.handler = companies_filter_by_sector
 
-    @dataclass(slots=True)
+    @dataclass
     class Company:
         name: str
         sector: str
@@ -29,5 +29,7 @@ def get_companies(
         companies = client.get_companies(sectors=["Information Technology", "Financials"])
 
     # THEN
-    assert len(companies) > 0
-    assert all(company.sector in ["Information Technology", "Financials"] for company in companies)
+    assert companies == [
+        Company(name="Apple", sector="Information Technology"),
+        Company(name="Berkshire Hathaway", sector="Financials"),
+    ]
diff --git a/tests/client/requests/test_array_query_params.py b/tests/client/requests/test_array_query_params.py
@@ -12,7 +12,7 @@ def test_send_array_query_params(http_server: HTTPTestServer) -> None:
     # GIVEN
     http_server.handler = companies_filter_by_sector
 
-    @dataclass(slots=True)
+    @dataclass
     class Company:
         name: str
         sector: str
@@ -29,5 +29,7 @@ def get_companies(
         companies = client.get_companies(sectors=["Information Technology", "Financials"])
 
     # THEN
-    assert len(companies) > 0
-    assert all(company.sector in ["Information Technology", "Financials"] for company in companies)
+    assert companies == [
+        Company(name="Apple", sector="Information Technology"),
+        Company(name="Berkshire Hathaway", sector="Financials"),
+    ]