Coverage for /var/srv/projects/api.amasfac.comuna18.com/tmp/venv/lib/python3.9/site-packages/pandas/_testing/compat.py: 23%

1""" 

2Helpers for sharing tests between DataFrame/Series 

3""" 

4 

5from pandas import DataFrame 

6 

7 

8def get_dtype(obj): 

9 if isinstance(obj, DataFrame): 

10 # Note: we are assuming only one column 

11 return obj.dtypes.iat[0] 

12 else: 

13 return obj.dtype 

14 

15 

16def get_obj(df: DataFrame, klass): 

17 """ 

18 For sharing tests using frame_or_series, either return the DataFrame 

19 unchanged or return it's first column as a Series. 

20 """ 

21 if klass is DataFrame: 

22 return df 

23 return df._ixs(0, axis=1)