tests: remove src from imports for tests

Author: IvanildoBarauna

.idea/api-to-dataframe.iml

@@ -1 +1 @@
-from src.api_to_dataframe.main import ClientBuilder
+from api_to_dataframe.main import ClientBuilder

src/api_to_dataframe/__init__.py

@@ -10,7 +10,7 @@ def __init__(self, endpoint: str):
             self.endpoint = endpoint
 
     def _response_to_json(self):
-        response = requests.get(self.url)
+        response = requests.get(self.endpoint)
 
         if response.ok:
             return response.json()
@@ -19,4 +19,4 @@ def _response_to_json(self):
 
     def to_dataframe(self):
         df = pd.DataFrame([self._response_to_json()])
-        return df
+        return df

src/api_to_dataframe/main.py

@@ -1,19 +1,32 @@
 import pytest
-import src.api_to_dataframe as apitodf
+import pandas as pd
+import api_to_dataframe as api_to_dataframe
+
+
+@pytest.fixture()
+def setup():
+    new_client = api_to_dataframe.ClientBuilder(endpoint="https://economia.awesomeapi.com.br/last/USD-BRL")
+    return new_client
 
 
 def test_constructor_without_param():
     with pytest.raises(ValueError):
-        new_client = apitodf.ClientBuilder(endpoint="")
+        new_client = api_to_dataframe.ClientBuilder(endpoint="")
 
 
-@pytest.fixture()
-def test_constructor_with_param():
-    expected_result = "param1"
-    new_client = apitodf.ClientBuilder(endpoint=expected_result)
+def test_constructor_with_param(setup):
+    expected_result = "https://economia.awesomeapi.com.br/last/USD-BRL"
+    new_client = setup
     assert new_client.endpoint == expected_result
-    return new_client
 
 
-def test_response_to_json_with_valid_endpoint(test_constructor_with_param):
-    pass
+def test_response_to_json(setup):
+    new_client = setup
+    response = new_client._response_to_json()
+    assert isinstance(response, dict)
+
+
+def test_to_dataframe(setup):
+    new_client = setup
+    df = new_client.to_dataframe()
+    assert isinstance(df, pd.DataFrame)