refactor: bump up version / fix import modules / add test

pyproject.toml

@@ -0,0 +1,4 @@
+[tool.pytest.ini_options]
+pythonpath = [
+  ".", "src", "tests", "src",
+]

setup.py

@@ -1,11 +1,12 @@
 import setuptools
+from src.version import __version__
 
 with open("README.md", "r") as ld:
     long_description = ld.read()
 
 setuptools.setup(
     name="myeia",
-    version="0.2.0",
+    version=__version__,
     url="https://github.com/philsv/myeia",
     license="MIT",
     author="philsv",

src/myeia.py

@@ -6,9 +6,8 @@
 import pandas as pd
 import requests
 from dotenv import load_dotenv
-from pandas.core.common import SettingWithCopyWarning
 
-warnings.simplefilter(action="ignore", category=SettingWithCopyWarning)
+warnings.simplefilter(action="ignore", category=pd.errors.PerformanceWarning)
 load_dotenv()
 
 
@@ -37,9 +36,7 @@ def get_data(
             "Accept": "*/*",
         }
 
-        api_route = (
-            f"{route}/data/?api_key={self.token}&data[]=value&frequency={frequency}"
-        )
+        api_route = f"{route}/data/?api_key={self.token}&data[]=value&frequency={frequency}"
 
         series = f"&facets[{facet}][]={series}"
 
@@ -73,15 +70,4 @@ def get_data(
         df.rename(columns={df.columns[0]: "Date"}, inplace=True)
         df["Date"] = pd.to_datetime(df["Date"])
         df.set_index("Date", inplace=True)
-
         return df
-
-
-# if __name__ == "__main__":
-
-#     # Example
-#     eia = API()
-#     df = eia.get_data(
-#         route="steo", series="PAPR_OPEC", frequency="monthly", facet="seriesId"
-#     )
-#     print(df.head())

src/version.py

@@ -1 +1 @@
-__version__ = "0.2.0"
+__version__ = "0.2.1"

tests/test_myeia.py

@@ -0,0 +1,9 @@
+import pandas as pd
+
+from src import myeia
+
+
+def test_get_data():
+    eia = myeia.API()
+    df = eia.get_data(route="natural-gas/pri/fut", series="RNGC1", frequency="daily")
+    assert isinstance(df, pd.DataFrame)