PYield 0.16.0__tar.gz → 0.16.2__tar.gz

{pyield-0.16.0 → pyield-0.16.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: PYield
-Version: 0.16.0
+Version: 0.16.2
 Summary: A Python library for analysis of fixed income instruments in Brazil
 Keywords: fixed-income, brazil, finance, analysis, bonds
 Author-Email: Carlos Carvalho <cr.cj@outlook.com>

pyield-0.16.2/pyield/__about__.py

@@ -0,0 +1 @@
+__version__ = "0.16.2"

{pyield-0.16.0 → pyield-0.16.2}/pyield/bday.py

@@ -116,11 +116,11 @@ def offset(
             `Series` of offset dates.
 
     Examples:
-        >>> date = "2023-12-23"  # Saturday before Christmas
+        >>> date = "23-12-2023"  # Saturday before Christmas
         >>> bday.offset(date, 0)
         Timestamp('2023-12-26')
 
-        >>> date = "2023-12-22"  # Friday before Christmas
+        >>> date = "22-12-2023"  # Friday before Christmas
         >>> bday.offset(date, 0)
         Timestamp('2023-12-22') # No offset because it's a business day
 

{pyield-0.16.0 → pyield-0.16.2}/pyield/bonds/ltn.py

@@ -152,7 +152,7 @@ def historical_premium(
     ltn_rates = df_anbima.query("MaturityDate == @maturity")["IndicativeRate"]
     if ltn_rates.empty:
         return float("NaN")
-    ltn_rate = ltn_rates.iloc[0]
+    ltn_rate = float(ltn_rates.iloc[0])
 
     # Retrieve DI rate for the reference date and maturity
     di_rate = di.rate(trade_date=reference_date, expiration=maturity)

{pyield-0.16.0 → pyield-0.16.2}/pyield/data/di.py

@@ -113,8 +113,6 @@ def rate(
     """
     # Convert input dates to consistent format
     trade_date = dc.convert_date(trade_date)
-    expiration = dc.convert_date(expiration)
-
     # Adjust expiration date to the nearest business day
     expiration = bday.offset(expiration, 0)
 
@@ -132,6 +130,10 @@ def rate(
     if df_exp.empty and not interpolate:
         return float("NaN")
 
+    if expiration in df_exp["ExpirationDate"]:
+        # Return the rate if an exact match is found
+        return float(df_exp["SettlementRate"].iloc[0])
+
     # Perform flat forward interpolation if required
     ff_interpolator = interpolator.Interpolator(
         method="flat_forward",
@@ -139,8 +141,5 @@ def rate(
         known_rates=df["SettlementRate"],
     )
 
-    # Calculate business days between trade date and expiration date
-    bdays = bday.count(trade_date, expiration)
-
     # Return the interpolated rate for the calculated business days
-    return ff_interpolator(bdays)
+    return ff_interpolator(bday.count(trade_date, expiration))

{pyield-0.16.0 → pyield-0.16.2}/pyield/interpolator.py

@@ -11,6 +11,7 @@ class Interpolator:
         method: Literal["flat_forward", "linear"],
         known_bdays: pd.Series | list,
         known_rates: pd.Series | list,
+        extrapolate: bool = False,
     ):
         """
         Initialize the Interpolator with given atributes.
@@ -19,6 +20,7 @@ class Interpolator:
             method (Literal["flat_forward", "linear"]): Interpolation method.
             known_bdays (pd.Series | pd.Index | list): Series of known business days.
             known_rates (pd.Series | pd.Index | list): Series of known interest rates.
+            extrapolate (bool, optional): Whether to extrapolate beyond the known data.
 
         Raises:
             ValueError: If known_bdays and known_rates do not have the same length.
@@ -43,6 +45,7 @@ class Interpolator:
         self.method = method
         self.known_bdays = known_bdays
         self.known_rates = known_rates
+        self.extrapolate = extrapolate
         self._set_known_bdays_and_rates()
 
     def _set_known_bdays_and_rates(self) -> None:
@@ -105,17 +108,18 @@ class Interpolator:
         # Check for edge cases
         if bday < self._known_bdays[0]:
             return self._known_rates[0]
-        elif bday > self._known_bdays[-1]:
-            return self._known_rates[-1]
         elif bday in self._known_bdays:
             return self._known_rates[self._known_bdays.index(bday)]
+        elif bday > self._known_bdays[-1]:
+            if self.extrapolate:
+                return self._known_rates[-1]
+            else:
+                return float("NaN")
 
         if self.method == "flat_forward":
             return self._flat_forward(bday)
         elif self.method == "linear":
             return self._linear(bday)
-        else:
-            raise ValueError(f"Unknown interpolation method: {self.method}.")
 
     def __call__(self, bday: int) -> float:
         """

{pyield-0.16.0 → pyield-0.16.2}/pyproject.toml

@@ -32,7 +32,7 @@ dependencies = [
     "types-requests",
 ]
 dynamic = []
-version = "0.16.0"
+version = "0.16.2"
 
 [project.license]
 file = "LICENSE"

pyield-0.16.0/pyield/__about__.py

@@ -1 +0,0 @@
-__version__ = "0.16.0"