refactor(endpoints): format known date columns in odata endpoints

Signed-off-by: Antonio Pedro Vieira <aps.vieira95@gmail.com>

Author: apsvieira

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):

pyproject.toml

@@ -24,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"

tests/test_expectativas.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)