Class CustomDataframe

lib/tests/streamlit/data_test_cases.py:112–121

A dummy dataframe-like class that supports the dataframe interchange protocol (__dataframe__ method).

class CustomDataframe:
    """A dummy dataframe-like class that supports the dataframe interchange protocol
    (__dataframe__ method).
    """

    def __init__(self, data: pd.DataFrame):
        self._data: pd.DataFrame = data

    def __dataframe__(self, allow_copy: bool = True):
        return self._data.__dataframe__(allow_copy=allow_copy)


class StrTestEnum(str, enum.Enum):