Merge pull request #10 from apsvieira/main

Adiciona parsing de colunas de data da API de expectativas

Author: wilsonfreitas

bcb/__init__.py

@@ -9,7 +9,6 @@
 )
 import pandas as pd
 
-
 OLINDA_BASE_URL = "https://olinda.bcb.gov.br/olinda/servico"
 
 
@@ -30,9 +29,19 @@ def __call__(self, *args):
 
 
 class EndpointQuery(ODataQuery):
+    _DATE_COLUMN_NAMES = {
+        "Data",
+        "DataReferencia",
+    }
+
     def collect(self):
-        data = super().collect()
-        return pd.DataFrame(data["value"])
+        raw_data = super().collect()
+        data = pd.DataFrame(raw_data["value"])
+        for col in self._DATE_COLUMN_NAMES:
+            if col not in data.columns:
+                continue
+            data[col] = pd.to_datetime(data[col])
+        return data
 
 
 class Endpoint(metaclass=EndpointMeta):

poetry.lock

@@ -15,6 +15,7 @@ requests = "^2.31.0"
 
 [tool.poetry.group.test.dependencies]
 pytest = "^7.1.3"
+flaky = "^3.7.0"
 
 [tool.poetry.group.docs.dependencies]
 Sphinx = "^5.1.1"
@@ -23,7 +24,6 @@ matplotlib = "^3.5.3"
 ipython = "^8.14.0"
 furo = "^2022.6.21"
 
-
 [tool.poetry.group.dev.dependencies]
 pycodestyle = "^2.9.1"
 ipykernel = "^6.15.2"

pyproject.toml

@@ -1,15 +1,14 @@
-import sys
-
-sys.path.append(".")
 from datetime import datetime, date
 import pandas as pd
+from pytest import mark
 from bcb import currency
 
 
 def test_currency_id():
     assert currency._get_currency_id("USD") == 61
 
 
+@mark.flaky(max_runs=20, min_passes=1)
 def test_currency_get_symbol():
     start_date = datetime.strptime("2020-12-01", "%Y-%m-%d")
     end_date = datetime.strptime("2020-12-05", "%Y-%m-%d")

tests/test_currency.py

@@ -0,0 +1,22 @@
+import pandas as pd
+from datetime import datetime
+from pytest import fixture
+
+from bcb import Expectativas
+
+
+@fixture
+def endpoints():
+    ep = Expectativas()
+    return [ep.get_endpoint(e.data["name"]) for e in ep.service.endpoints]
+
+
+def test_expectativas_date_format(endpoints):
+    for endpoint in endpoints:
+        query = endpoint.query().limit(1)
+        data = query.collect()
+        assert isinstance(data, pd.DataFrame)
+        assert data.shape[0] == 1
+        assert isinstance(data["Data"].iloc[0], datetime)
+        if "DataReferencia" in data.columns:
+            assert isinstance(data["DataReferencia"].iloc[0], datetime)

tests/test_expectativas.py

@@ -1,6 +1,3 @@
-import sys
-
-sys.path.append(".")
 from datetime import datetime
 import pandas as pd
 from bcb import sgs

tests/test_series.py

@@ -1,7 +1,3 @@
-import sys
-
-sys.path.append(".")
-
 from datetime import datetime, date
 
 from bcb import utils