44 lines
1.0 KiB
Python
44 lines
1.0 KiB
Python
import pytest
|
|
|
|
import pandas as pd
|
|
from pandas.core.arrays.floating import (
|
|
Float32Dtype,
|
|
Float64Dtype,
|
|
)
|
|
|
|
|
|
@pytest.fixture(params=[Float32Dtype, Float64Dtype])
|
|
def dtype(request):
|
|
"""Parametrized fixture returning a float 'dtype'"""
|
|
return request.param()
|
|
|
|
|
|
@pytest.fixture
|
|
def data(dtype):
|
|
"""Fixture returning 'data' array according to parametrized float 'dtype'"""
|
|
return pd.array(
|
|
[0.1, 0.2, 0.3, 0.4, pd.NA, 1.0, 1.1, pd.NA, 9.9, 10.0],
|
|
dtype=dtype,
|
|
)
|
|
|
|
|
|
@pytest.fixture
|
|
def data_missing(dtype):
|
|
"""
|
|
Fixture returning array with missing data according to parametrized float
|
|
'dtype'.
|
|
"""
|
|
return pd.array([pd.NA, 0.1], dtype=dtype)
|
|
|
|
|
|
@pytest.fixture(params=["data", "data_missing"])
|
|
def all_data(request, data, data_missing):
|
|
"""Parametrized fixture returning 'data' or 'data_missing' float arrays.
|
|
|
|
Used to test dtype conversion with and without missing values.
|
|
"""
|
|
if request.param == "data":
|
|
return data
|
|
elif request.param == "data_missing":
|
|
return data_missing
|