Coverage for src/ipyvizzu/data/converters/pandas/protocol.py: 100%

1""" 

2This module provides protocol classes for pandas data frame converter. 

3""" 

4 

5from typing import Any, Callable, Sequence 

6 

7from ipyvizzu.__version__ import PYENV 

8 

9 

10if PYENV >= (3, 8): 

11 from typing import Protocol, runtime_checkable 

12else: 

13 # TODO: remove once support for Python 3.7 is dropped 

14 # pylint: disable=duplicate-code 

15 from typing_extensions import Protocol, runtime_checkable # type: ignore 

16 

17 

18@runtime_checkable 

19class PandasDataFrame(Protocol): 

20 """ 

21 Represents a pandas DataFrame Protocol. 

22 """ 

23 

24 # pylint: disable=too-few-public-methods 

25 

26 index: Any 

27 columns: Sequence[str] 

28 sample: Callable[..., Any] 

29 __len__: Callable[[], int] 

30 __getitem__: Callable[[Any], Any] 

31 

32 

33@runtime_checkable 

34class PandasSeries(Protocol): 

35 """ 

36 Represents a pandas Series Protocol. 

37 """ 

38 

39 # pylint: disable=too-few-public-methods 

40 

41 index: Any 

42 values: Any 

43 dtype: Any 

44 __len__: Callable[[], int] 

45 __getitem__: Callable[[Any], Any]