Inzynierka/Lib/site-packages/pandas/tests/util/test_validate_inclusive.py
2023-06-02 12:51:02 +02:00

41 lines
896 B
Python

import numpy as np
import pytest
from pandas.util._validators import validate_inclusive
import pandas as pd
@pytest.mark.parametrize(
"invalid_inclusive",
(
"ccc",
2,
object(),
None,
np.nan,
pd.NA,
pd.DataFrame(),
),
)
def test_invalid_inclusive(invalid_inclusive):
with pytest.raises(
ValueError,
match="Inclusive has to be either 'both', 'neither', 'left' or 'right'",
):
validate_inclusive(invalid_inclusive)
@pytest.mark.parametrize(
"valid_inclusive, expected_tuple",
(
("left", (True, False)),
("right", (False, True)),
("both", (True, True)),
("neither", (False, False)),
),
)
def test_valid_inclusive(valid_inclusive, expected_tuple):
resultant_tuple = validate_inclusive(valid_inclusive)
assert expected_tuple == resultant_tuple