s

Author: afuetterer

tests/integration/test_async_client.py

@@ -68,15 +68,16 @@ async def test_client_list_repositories_dict(async_client: AsyncClient, mock_rep
 async def test_client_list_repositories_csv(async_client: AsyncClient, mock_repository_list_route: Route) -> None:
     response = await async_client.repositories.list(return_type=ReturnType.CSV)
     assert isinstance(response, str)
-    assert '"id": "r3d100010371",' in response
-    assert '"doi": "https://doi.org/10.17616/R3P594",' in response
+    assert response.startswith("id,doi,name,")
+    assert "r3d100010371" in response
+    assert "https://doi.org/10.17616/R3P594" in response
 
 
 async def test_client_list_repositories_dataframe(async_client: AsyncClient, mock_repository_list_route: Route) -> None:
     response = await async_client.repositories.list(return_type=ReturnType.DATAFRAME)
-    assert isinstance(response, str)
-    assert '"id": "r3d100010371",' in response
-    assert '"doi": "https://doi.org/10.17616/R3P594",' in response
+    assert isinstance(response, DataFrame)
+    assert response.shape == (3, 5)
+    assert response["id"].loc[0] == "r3d100010371"
 
 
 async def test_client_list_repositories_response(async_client: AsyncClient, mock_repository_list_route: Route) -> None:
@@ -159,17 +160,17 @@ async def test_client_get_single_repository_csv(
 ) -> None:
     response = await async_client.repositories.get(zenodo_id, return_type=ReturnType.CSV)
     assert isinstance(response, str)
-    assert "id,doi,name,link.href" in response
-    assert '"re3data.orgIdentifier": "r3d100010468",' in response
+    assert response.startswith("re3data.orgIdentifier,additionalName,repositoryURL,")
+    assert "r3d100010468" in response
 
 
 async def test_client_get_single_repository_dataframe(
     async_client: AsyncClient, mock_repository_get_route: Route, zenodo_id: str
 ) -> None:
     response = await async_client.repositories.get(zenodo_id, return_type=ReturnType.DATAFRAME)
     assert isinstance(response, DataFrame)
-    assert "{" in response
-    assert '"re3data.orgIdentifier": "r3d100010468",' in response
+    assert response.shape == (1, 43)
+    assert response["re3data.orgIdentifier"].loc[0] == "r3d100010468"
 
 
 async def test_client_get_single_repository_response(

tests/integration/test_client.py

@@ -68,15 +68,16 @@ def test_client_list_repositories_dict(client: Client, mock_repository_list_rout
 def test_client_list_repositories_csv(client: Client, mock_repository_list_route: Route) -> None:
     response = client.repositories.list(return_type=ReturnType.CSV)
     assert isinstance(response, str)
-    assert '"id": "r3d100010371",' in response
-    assert '"doi": "https://doi.org/10.17616/R3P594",' in response
+    assert response.startswith("id,doi,name,")
+    assert "r3d100010371" in response
+    assert "https://doi.org/10.17616/R3P594" in response
 
 
 def test_client_list_repositories_dataframe(client: Client, mock_repository_list_route: Route) -> None:
     response = client.repositories.list(return_type=ReturnType.DATAFRAME)
-    assert isinstance(response, str)
-    assert '"id": "r3d100010371",' in response
-    assert '"doi": "https://doi.org/10.17616/R3P594",' in response
+    assert isinstance(response, DataFrame)
+    assert response.shape == (3, 5)
+    assert response["id"].loc[0] == "r3d100010371"
 
 
 def test_client_list_repositories_response(client: Client, mock_repository_list_route: Route) -> None:
@@ -149,17 +150,17 @@ def test_client_get_single_repository_dict(client: Client, mock_repository_get_r
 def test_client_get_single_repository_csv(client: Client, mock_repository_get_route: Route, zenodo_id: str) -> None:
     response = client.repositories.get(zenodo_id, return_type=ReturnType.CSV)
     assert isinstance(response, str)
-    assert "id,doi,name,link.href" in response
-    assert '"re3data.orgIdentifier": "r3d100010468",' in response
+    assert response.startswith("re3data.orgIdentifier,additionalName,repositoryURL,")
+    assert "r3d100010468" in response
 
 
 def test_client_get_single_repository_dataframe(
     client: Client, mock_repository_get_route: Route, zenodo_id: str
 ) -> None:
     response = client.repositories.get(zenodo_id, return_type=ReturnType.DATAFRAME)
     assert isinstance(response, DataFrame)
-    assert "{" in response
-    assert '"re3data.orgIdentifier": "r3d100010468",' in response
+    assert response.shape == (1, 43)
+    assert response["re3data.orgIdentifier"].loc[0] == "r3d100010468"
 
 
 def test_client_get_single_repository_response(