finbrain-python 0.2.0__tar.gz → 0.2.1__tar.gz

{finbrain_python-0.2.0 → finbrain_python-0.2.1}/CHANGELOG.md

@@ -5,6 +5,15 @@ All notable changes to this project will be documented in this file.
 The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/),
 and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 
+## [0.2.1] - 2026-03-12
+
+### Added
+
+- **Corporate Lobbying Endpoint**: `fb.corporate_lobbying.ticker("AAPL")` — fetch corporate lobbying filings with income, expenses, registrant details, issue codes, and government entities (`/lobbying/{SYMBOL}`)
+- **Corporate Lobbying Plotting**: `fb.plot.corporate_lobbying()` — visualize lobbying spend (income + expenses) as bars on a secondary y-axis overlaid on a price chart, with hover details showing registrant, quarter, income, and expenses
+- **Async Corporate Lobbying**: Full async support via `AsyncCorporateLobbyingAPI`
+- **Corporate Lobbying Tests**: Unit tests (`tests/test_corporate_lobbying.py`) and integration tests
+
 ## [0.2.0] - 2026-03-09
 
 ### BREAKING CHANGES
@@ -128,6 +137,7 @@ Previous releases...
 
 Previous releases...
 
+[0.2.1]: https://github.com/ahmetsbilgin/finbrain-python/compare/v0.2.0...v0.2.1
 [0.2.0]: https://github.com/ahmetsbilgin/finbrain-python/compare/v0.1.8...v0.2.0
 [0.1.8]: https://github.com/ahmetsbilgin/finbrain-python/compare/v0.1.7...v0.1.8
 [0.1.7]: https://github.com/ahmetsbilgin/finbrain-python/compare/v0.1.6...v0.1.7

{finbrain_python-0.2.0 → finbrain_python-0.2.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: finbrain-python
-Version: 0.2.0
+Version: 0.2.1
 Summary: Official Python client for the FinBrain API
 Author-email: Ahmet Salim Bilgin <ahmet@finbrain.tech>
 License-Expression: MIT
@@ -93,6 +93,12 @@ fb.senate_trades.ticker("META",
                         date_to="2025-06-30",
                         as_dataframe=True)
 
+# ---------- corporate lobbying ----------
+fb.corporate_lobbying.ticker("AAPL",
+                             date_from="2024-01-01",
+                             date_to="2025-06-30",
+                             as_dataframe=True)
+
 # ---------- insider transactions ----------
 fb.insider_transactions.ticker("AMZN", as_dataframe=True)
 
@@ -205,7 +211,7 @@ fb.plot.sentiments("AMZN",
                    date_from="2025-01-01",
                    date_to="2025-06-30")
 
-# ---------- Insider Transactions, House & Senate Trades (requires user price data) ----------
+# ---------- Insider Transactions, House & Senate Trades, Corporate Lobbying (requires user price data) ----------
 # These plots overlay transaction markers on a price chart.
 # Since FinBrain doesn't provide historical prices, you must provide your own:
 
@@ -232,6 +238,12 @@ fb.plot.senate_trades("META",
                       price_data=price_df,
                       date_from="2025-01-01",
                       date_to="2025-06-30")
+
+# Plot corporate lobbying spend on your price chart
+fb.plot.corporate_lobbying("AAPL",
+                           price_data=price_df,
+                           date_from="2024-01-01",
+                           date_to="2025-06-30")
 ```
 
 **Price Data Requirements:**
@@ -276,6 +288,7 @@ fb = FinBrainClient()  # reads from FINBRAIN_API_KEY env var
 | Analyst ratings      | `client.analyst_ratings.ticker()`        | `/analyst-ratings/{SYMBOL}`                 |
 | House trades         | `client.house_trades.ticker()`           | `/congress/house/{SYMBOL}`                  |
 | Senate trades        | `client.senate_trades.ticker()`          | `/congress/senate/{SYMBOL}`                 |
+| Corporate lobbying   | `client.corporate_lobbying.ticker()`     | `/lobbying/{SYMBOL}`                        |
 | Insider transactions | `client.insider_transactions.ticker()`   | `/insider-trading/{SYMBOL}`                 |
 | LinkedIn             | `client.linkedin_data.ticker()`          | `/linkedin/{SYMBOL}`                        |
 | Options – Put/Call   | `client.options.put_call()`              | `/put-call-ratio/{SYMBOL}`                  |

{finbrain_python-0.2.0 → finbrain_python-0.2.1}/README.md

@@ -67,6 +67,12 @@ fb.senate_trades.ticker("META",
                         date_to="2025-06-30",
                         as_dataframe=True)
 
+# ---------- corporate lobbying ----------
+fb.corporate_lobbying.ticker("AAPL",
+                             date_from="2024-01-01",
+                             date_to="2025-06-30",
+                             as_dataframe=True)
+
 # ---------- insider transactions ----------
 fb.insider_transactions.ticker("AMZN", as_dataframe=True)
 
@@ -179,7 +185,7 @@ fb.plot.sentiments("AMZN",
                    date_from="2025-01-01",
                    date_to="2025-06-30")
 
-# ---------- Insider Transactions, House & Senate Trades (requires user price data) ----------
+# ---------- Insider Transactions, House & Senate Trades, Corporate Lobbying (requires user price data) ----------
 # These plots overlay transaction markers on a price chart.
 # Since FinBrain doesn't provide historical prices, you must provide your own:
 
@@ -206,6 +212,12 @@ fb.plot.senate_trades("META",
                       price_data=price_df,
                       date_from="2025-01-01",
                       date_to="2025-06-30")
+
+# Plot corporate lobbying spend on your price chart
+fb.plot.corporate_lobbying("AAPL",
+                           price_data=price_df,
+                           date_from="2024-01-01",
+                           date_to="2025-06-30")
 ```
 
 **Price Data Requirements:**
@@ -250,6 +262,7 @@ fb = FinBrainClient()  # reads from FINBRAIN_API_KEY env var
 | Analyst ratings      | `client.analyst_ratings.ticker()`        | `/analyst-ratings/{SYMBOL}`                 |
 | House trades         | `client.house_trades.ticker()`           | `/congress/house/{SYMBOL}`                  |
 | Senate trades        | `client.senate_trades.ticker()`          | `/congress/senate/{SYMBOL}`                 |
+| Corporate lobbying   | `client.corporate_lobbying.ticker()`     | `/lobbying/{SYMBOL}`                        |
 | Insider transactions | `client.insider_transactions.ticker()`   | `/insider-trading/{SYMBOL}`                 |
 | LinkedIn             | `client.linkedin_data.ticker()`          | `/linkedin/{SYMBOL}`                        |
 | Options – Put/Call   | `client.options.put_call()`              | `/put-call-ratio/{SYMBOL}`                  |

{finbrain_python-0.2.0 → finbrain_python-0.2.1}/src/finbrain/aio/client.py

@@ -22,6 +22,7 @@ from .endpoints.options import AsyncOptionsAPI
 from .endpoints.news import AsyncNewsAPI
 from .endpoints.screener import AsyncScreenerAPI
 from .endpoints.recent import AsyncRecentAPI
+from .endpoints.corporate_lobbying import AsyncCorporateLobbyingAPI
 
 
 # Which status codes merit a retry
@@ -80,6 +81,7 @@ class AsyncFinBrainClient:
         self.news = AsyncNewsAPI(self)
         self.screener = AsyncScreenerAPI(self)
         self.recent = AsyncRecentAPI(self)
+        self.corporate_lobbying = AsyncCorporateLobbyingAPI(self)
 
     async def __aenter__(self) -> "AsyncFinBrainClient":
         """Context manager entry."""

finbrain_python-0.2.1/src/finbrain/aio/endpoints/corporate_lobbying.py

@@ -0,0 +1,48 @@
+from __future__ import annotations
+import pandas as pd
+import datetime as _dt
+from typing import TYPE_CHECKING, Dict, Any
+
+from ._utils import to_datestr
+
+if TYPE_CHECKING:
+    from ..client import AsyncFinBrainClient
+
+
+class AsyncCorporateLobbyingAPI:
+    """Async wrapper for /lobbying endpoints."""
+
+    def __init__(self, client: "AsyncFinBrainClient") -> None:
+        self._c = client
+
+    async def ticker(
+        self,
+        symbol: str,
+        *,
+        date_from: _dt.date | str | None = None,
+        date_to: _dt.date | str | None = None,
+        limit: int | None = None,
+        as_dataframe: bool = False,
+    ) -> Dict[str, Any] | pd.DataFrame:
+        """Fetch corporate lobbying filings for a symbol (async)."""
+        params: Dict[str, str] = {}
+        if date_from:
+            params["startDate"] = to_datestr(date_from)
+        if date_to:
+            params["endDate"] = to_datestr(date_to)
+        if limit is not None:
+            params["limit"] = str(limit)
+
+        path = f"lobbying/{symbol.upper()}"
+
+        data: Dict[str, Any] = await self._c._request("GET", path, params=params)
+
+        if as_dataframe:
+            rows = data.get("filings", [])
+            df = pd.DataFrame(rows)
+            if not df.empty and "date" in df.columns:
+                df["date"] = pd.to_datetime(df["date"])
+                df.set_index("date", inplace=True)
+            return df
+
+        return data

{finbrain_python-0.2.0 → finbrain_python-0.2.1}/src/finbrain/client.py

@@ -23,6 +23,7 @@ from .endpoints.options import OptionsAPI
 from .endpoints.news import NewsAPI
 from .endpoints.screener import ScreenerAPI
 from .endpoints.recent import RecentAPI
+from .endpoints.corporate_lobbying import CorporateLobbyingAPI
 
 
 # Which status codes merit a retry
@@ -75,6 +76,7 @@ class FinBrainClient:
         self.news = NewsAPI(self)
         self.screener = ScreenerAPI(self)
         self.recent = RecentAPI(self)
+        self.corporate_lobbying = CorporateLobbyingAPI(self)
 
     # ---------- private helpers ----------
     def _request(

finbrain_python-0.2.1/src/finbrain/endpoints/corporate_lobbying.py

@@ -0,0 +1,72 @@
+from __future__ import annotations
+import pandas as pd
+import datetime as _dt
+from typing import TYPE_CHECKING, Dict, Any, List
+
+from ._utils import to_datestr
+
+if TYPE_CHECKING:  # imported only by type-checkers
+    from ..client import FinBrainClient
+
+
+class CorporateLobbyingAPI:
+    """
+    Endpoint
+    --------
+    ``/lobbying/<TICKER>`` - corporate lobbying filings for the selected ticker.
+    """
+
+    # ------------------------------------------------------------------ #
+    def __init__(self, client: "FinBrainClient") -> None:
+        self._c = client  # reference to the parent client
+
+    # ------------------------------------------------------------------ #
+    def ticker(
+        self,
+        symbol: str,
+        *,
+        date_from: _dt.date | str | None = None,
+        date_to: _dt.date | str | None = None,
+        limit: int | None = None,
+        as_dataframe: bool = False,
+    ) -> Dict[str, Any] | pd.DataFrame:
+        """
+        Fetch corporate lobbying filings for *symbol*.
+
+        Parameters
+        ----------
+        symbol :
+            Ticker symbol; auto-upper-cased.
+        date_from, date_to :
+            Optional ISO dates (``YYYY-MM-DD``) bounding the returned rows.
+        limit :
+            Maximum number of records to return.
+        as_dataframe :
+            If *True*, return a **pandas.DataFrame** indexed by ``date``;
+            otherwise return the raw JSON dict.
+
+        Returns
+        -------
+        dict | pandas.DataFrame
+        """
+        params: Dict[str, str] = {}
+        if date_from:
+            params["startDate"] = to_datestr(date_from)
+        if date_to:
+            params["endDate"] = to_datestr(date_to)
+        if limit is not None:
+            params["limit"] = str(limit)
+
+        path = f"lobbying/{symbol.upper()}"
+
+        data: Dict[str, Any] = self._c._request("GET", path, params=params)
+
+        if as_dataframe:
+            rows: List[Dict[str, Any]] = data.get("filings", [])
+            df = pd.DataFrame(rows)
+            if not df.empty and "date" in df.columns:
+                df["date"] = pd.to_datetime(df["date"])
+                df.set_index("date", inplace=True)
+            return df
+
+        return data

{finbrain_python-0.2.0 → finbrain_python-0.2.1}/src/finbrain/plotting.py

@@ -751,6 +751,162 @@ class _PlotNamespace:
             return None
         return fig.to_json() if as_json else fig
 
+    # --------------------------------------------------------------------- #
+    # Corporate Lobbying  → bars on price chart                              #
+    # --------------------------------------------------------------------- #
+    def corporate_lobbying(
+        self,
+        ticker: str,
+        price_data: pd.DataFrame,
+        *,
+        date_from: str | None = None,
+        date_to: str | None = None,
+        as_json: bool = False,
+        show: bool = True,
+        template: str = "plotly_dark",
+        **kwargs,
+    ):
+        """
+        Plot corporate lobbying spend overlaid on a price chart.
+
+        This method requires user-provided historical price data, as FinBrain
+        does not currently offer a price history endpoint.
+
+        Parameters
+        ----------
+        ticker : str
+            Ticker symbol (e.g. ``"AAPL"``).
+        price_data : pandas.DataFrame
+            **User-provided** price history with a DatetimeIndex and a column
+            containing prices (e.g. ``"close"``, ``"Close"``, or ``"price"``).
+            The index must be timezone-naive or UTC.
+        date_from, date_to : str or None, optional
+            Date range for filings in ``YYYY-MM-DD`` format.
+        as_json : bool, default False
+            If ``True``, return JSON string instead of Figure object.
+        show : bool, default True
+            If ``True`` and ``as_json=False``, display the figure immediately.
+        template : str, default "plotly_dark"
+            Plotly template name.
+        **kwargs
+            Additional arguments passed to
+            :meth:`FinBrainClient.corporate_lobbying.ticker`.
+
+        Returns
+        -------
+        plotly.graph_objects.Figure or str or None
+            Figure object, JSON string, or None (when shown).
+
+        Raises
+        ------
+        ValueError
+            If ``price_data`` is empty or missing required price column.
+        """
+        # Validate price_data
+        if price_data.empty:
+            raise ValueError("price_data cannot be empty")
+
+        # Flatten MultiIndex columns if present (e.g., from yf.download())
+        if isinstance(price_data.columns, pd.MultiIndex):
+            price_data = price_data.copy()
+            price_data.columns = price_data.columns.get_level_values(0)
+
+        # Find price column (case-insensitive search)
+        price_col = None
+        for col in ["close", "Close", "price", "Price", "adj_close", "Adj Close"]:
+            if col in price_data.columns:
+                price_col = col
+                break
+        if price_col is None:
+            raise ValueError(
+                f"price_data must contain a price column (e.g. 'close', 'Close', 'price'). "
+                f"Found columns: {price_data.columns.tolist()}"
+            )
+
+        # Fetch lobbying filings
+        filings_df = self._fb.corporate_lobbying.ticker(
+            ticker,
+            date_from=date_from,
+            date_to=date_to,
+            as_dataframe=True,
+            **kwargs,
+        )
+
+        # Normalize timezones
+        price_data_normalized = price_data.copy()
+        if price_data_normalized.index.tz is not None:
+            price_data_normalized.index = price_data_normalized.index.tz_localize(None)
+
+        fig = go.Figure(
+            layout=dict(
+                template=template,
+                title=f"Corporate Lobbying · {ticker}",
+                xaxis_title="Date",
+                hovermode="x unified",
+            )
+        )
+
+        # Plot price line on primary y-axis
+        fig.add_scatter(
+            name="Price",
+            x=price_data_normalized.index,
+            y=price_data_normalized[price_col],
+            mode="lines",
+            line=dict(width=2, color="#02d2ff"),
+            hovertemplate="<b>%{x|%Y-%m-%d}</b><br>Price: $%{y:.2f}<extra></extra>",
+        )
+
+        if not filings_df.empty:
+            filings_normalized = filings_df.copy()
+            if filings_normalized.index.tz is not None:
+                filings_normalized.index = filings_normalized.index.tz_localize(None)
+
+            # Compute total spend per filing (income + expenses)
+            spend = filings_normalized.get("income", 0) + filings_normalized.get(
+                "expenses", 0
+            )
+
+            hover_text = []
+            for _, row in filings_normalized.iterrows():
+                registrant = row.get("registrantName", "N/A")
+                quarter = row.get("quarter", "")
+                income = row.get("income", 0)
+                expenses = row.get("expenses", 0)
+                hover_text.append(
+                    f"Registrant: {registrant}<br>"
+                    f"Quarter: {quarter}<br>"
+                    f"Income: ${income:,.0f}<br>"
+                    f"Expenses: ${expenses:,.0f}"
+                )
+
+            # Plot lobbying spend as bars on secondary y-axis
+            fig.add_bar(
+                name="Lobbying Spend",
+                x=filings_normalized.index,
+                y=spend,
+                marker_color="rgba(249,200,14,0.6)",
+                yaxis="y2",
+                hovertext=hover_text,
+                hovertemplate="<b>%{x|%Y-%m-%d}</b><br>%{hovertext}<extra></extra>",
+            )
+
+        fig.update_layout(
+            yaxis=dict(title="Price", showgrid=True),
+            yaxis2=dict(
+                title="Lobbying Spend ($)",
+                overlaying="y",
+                side="right",
+                showgrid=False,
+                zeroline=False,
+                rangemode="tozero",
+            ),
+        )
+
+        if show and not as_json:
+            fig.show()
+            return None
+        return fig.to_json() if as_json else fig
+
     # --------------------------------------------------------------------- #
     # Helper methods                                                         #
     # --------------------------------------------------------------------- #

{finbrain_python-0.2.0 → finbrain_python-0.2.1}/src/finbrain_python.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: finbrain-python
-Version: 0.2.0
+Version: 0.2.1
 Summary: Official Python client for the FinBrain API
 Author-email: Ahmet Salim Bilgin <ahmet@finbrain.tech>
 License-Expression: MIT
@@ -93,6 +93,12 @@ fb.senate_trades.ticker("META",
                         date_to="2025-06-30",
                         as_dataframe=True)
 
+# ---------- corporate lobbying ----------
+fb.corporate_lobbying.ticker("AAPL",
+                             date_from="2024-01-01",
+                             date_to="2025-06-30",
+                             as_dataframe=True)
+
 # ---------- insider transactions ----------
 fb.insider_transactions.ticker("AMZN", as_dataframe=True)
 
@@ -205,7 +211,7 @@ fb.plot.sentiments("AMZN",
                    date_from="2025-01-01",
                    date_to="2025-06-30")
 
-# ---------- Insider Transactions, House & Senate Trades (requires user price data) ----------
+# ---------- Insider Transactions, House & Senate Trades, Corporate Lobbying (requires user price data) ----------
 # These plots overlay transaction markers on a price chart.
 # Since FinBrain doesn't provide historical prices, you must provide your own:
 
@@ -232,6 +238,12 @@ fb.plot.senate_trades("META",
                       price_data=price_df,
                       date_from="2025-01-01",
                       date_to="2025-06-30")
+
+# Plot corporate lobbying spend on your price chart
+fb.plot.corporate_lobbying("AAPL",
+                           price_data=price_df,
+                           date_from="2024-01-01",
+                           date_to="2025-06-30")
 ```
 
 **Price Data Requirements:**
@@ -276,6 +288,7 @@ fb = FinBrainClient()  # reads from FINBRAIN_API_KEY env var
 | Analyst ratings      | `client.analyst_ratings.ticker()`        | `/analyst-ratings/{SYMBOL}`                 |
 | House trades         | `client.house_trades.ticker()`           | `/congress/house/{SYMBOL}`                  |
 | Senate trades        | `client.senate_trades.ticker()`          | `/congress/senate/{SYMBOL}`                 |
+| Corporate lobbying   | `client.corporate_lobbying.ticker()`     | `/lobbying/{SYMBOL}`                        |
 | Insider transactions | `client.insider_transactions.ticker()`   | `/insider-trading/{SYMBOL}`                 |
 | LinkedIn             | `client.linkedin_data.ticker()`          | `/linkedin/{SYMBOL}`                        |
 | Options – Put/Call   | `client.options.put_call()`              | `/put-call-ratio/{SYMBOL}`                  |

{finbrain_python-0.2.0 → finbrain_python-0.2.1}/src/finbrain_python.egg-info/SOURCES.txt

@@ -21,6 +21,7 @@ src/finbrain/aio/endpoints/_utils.py
 src/finbrain/aio/endpoints/analyst_ratings.py
 src/finbrain/aio/endpoints/app_ratings.py
 src/finbrain/aio/endpoints/available.py
+src/finbrain/aio/endpoints/corporate_lobbying.py
 src/finbrain/aio/endpoints/house_trades.py
 src/finbrain/aio/endpoints/insider_transactions.py
 src/finbrain/aio/endpoints/linkedin_data.py
@@ -36,6 +37,7 @@ src/finbrain/endpoints/_utils.py
 src/finbrain/endpoints/analyst_ratings.py
 src/finbrain/endpoints/app_ratings.py
 src/finbrain/endpoints/available.py
+src/finbrain/endpoints/corporate_lobbying.py
 src/finbrain/endpoints/house_trades.py
 src/finbrain/endpoints/insider_transactions.py
 src/finbrain/endpoints/linkedin_data.py
@@ -57,6 +59,7 @@ tests/test_analyst_ratings.py
 tests/test_app_ratings.py
 tests/test_async_client.py
 tests/test_available.py
+tests/test_corporate_lobbying.py
 tests/test_envelope.py
 tests/test_house_trades.py
 tests/test_insider_transactions.py

{finbrain_python-0.2.0 → finbrain_python-0.2.1}/tests/test_async_client.py

@@ -71,3 +71,4 @@ async def test_async_endpoints_exist():
         assert hasattr(client, "insider_transactions")
         assert hasattr(client, "linkedin_data")
         assert hasattr(client, "options")
+        assert hasattr(client, "corporate_lobbying")

finbrain_python-0.2.1/tests/test_corporate_lobbying.py

@@ -0,0 +1,104 @@
+# tests/test_corporate_lobbying.py
+import pandas as pd
+import pytest
+
+from .conftest import stub_json, wrap_v2
+from finbrain.exceptions import BadRequest
+
+TICKER = "AAPL"
+
+
+# ─────────── raw JSON branch ────────────────────────────────────────────
+def test_corporate_lobbying_raw_ok(client, _activate_responses):
+    path = f"lobbying/{TICKER}"
+    payload = wrap_v2({
+        "symbol": "AAPL",
+        "name": "Apple Inc.",
+        "filings": [
+            {
+                "date": "2024-03-15",
+                "filingUuid": "abc-123",
+                "filingYear": 2024,
+                "quarter": "Q1",
+                "clientName": "Apple Inc.",
+                "registrantName": "Lobbying Firm LLC",
+                "income": 50000,
+                "expenses": 75000,
+                "issueCodes": ["TAX", "ITC"],
+                "governmentEntities": ["Congress", "Senate"],
+            }
+        ],
+    })
+
+    stub_json(_activate_responses, "GET", path, payload)
+
+    data = client.corporate_lobbying.ticker(symbol=TICKER)
+    assert data["symbol"] == TICKER
+    assert isinstance(data["filings"], list)
+    assert data["filings"][0]["registrantName"] == "Lobbying Firm LLC"
+
+
+# ─────────── DataFrame branch ───────────────────────────────────────────
+def test_corporate_lobbying_dataframe_ok(client, _activate_responses):
+    path = f"lobbying/{TICKER}"
+    payload = wrap_v2({
+        "symbol": "AAPL",
+        "name": "Apple Inc.",
+        "filings": [
+            {
+                "date": "2024-03-15",
+                "filingUuid": "abc-123",
+                "filingYear": 2024,
+                "quarter": "Q1",
+                "clientName": "Apple Inc.",
+                "registrantName": "Lobbying Firm LLC",
+                "income": 50000,
+                "expenses": 75000,
+                "issueCodes": ["TAX", "ITC"],
+                "governmentEntities": ["Congress", "Senate"],
+            },
+            {
+                "date": "2024-06-20",
+                "filingUuid": "def-456",
+                "filingYear": 2024,
+                "quarter": "Q2",
+                "clientName": "Apple Inc.",
+                "registrantName": "Another Firm Inc.",
+                "income": 30000,
+                "expenses": 45000,
+                "issueCodes": ["ENV"],
+                "governmentEntities": ["Congress"],
+            },
+        ],
+    })
+
+    stub_json(_activate_responses, "GET", path, payload)
+
+    df = client.corporate_lobbying.ticker(symbol=TICKER, as_dataframe=True)
+
+    assert isinstance(df, pd.DataFrame)
+    assert len(df) == 2
+    assert "registrantName" in df.columns
+    assert "income" in df.columns
+    assert "expenses" in df.columns
+    assert df.index.name == "date"
+    assert pd.api.types.is_datetime64_any_dtype(df.index)
+    assert pd.Timestamp("2024-03-15") in df.index
+    assert pd.Timestamp("2024-06-20") in df.index
+    assert df.loc["2024-03-15", "registrantName"] == "Lobbying Firm LLC"
+    assert df.loc["2024-06-20", "income"] == 30000
+
+
+# ─────────── error mapping ──────────────────────────────────────────────
+def test_corporate_lobbying_bad_request(client, _activate_responses):
+    path = f"lobbying/{TICKER}"
+    stub_json(
+        _activate_responses,
+        "GET",
+        path,
+        {"success": False, "error": {"code": "VALIDATION_ERROR", "message": "bad"}},
+        status=400,
+    )
+
+    with pytest.raises(BadRequest):
+        client.corporate_lobbying.ticker(symbol=TICKER)

{finbrain_python-0.2.0 → finbrain_python-0.2.1}/tests/test_integration.py

@@ -261,6 +261,33 @@ class TestSenateTrades:
             assert "politician" in df.columns
 
 
+# =====================================================================
+# Corporate Lobbying
+# =====================================================================
+class TestCorporateLobbying:
+    def test_raw(self, fb):
+        data = fb.corporate_lobbying.ticker(TICKER)
+        assert isinstance(data, dict)
+        assert "filings" in data
+        filings = data["filings"]
+        assert isinstance(filings, list)
+        if len(filings) > 0:
+            assert "date" in filings[0]
+            assert "registrantName" in filings[0]
+            assert "income" in filings[0]
+            assert "expenses" in filings[0]
+
+    def test_dataframe(self, fb):
+        df = fb.corporate_lobbying.ticker(TICKER, as_dataframe=True)
+        assert isinstance(df, pd.DataFrame)
+        if len(df) > 0:
+            assert df.index.name == "date"
+            assert pd.api.types.is_datetime64_any_dtype(df.index)
+            assert "registrantName" in df.columns
+            assert "income" in df.columns
+            assert "expenses" in df.columns
+
+
 # =====================================================================
 # LinkedIn Data
 # =====================================================================