FM / indexing

/usr/local/lib64/python3.6/site-packages/pandas/tests/indexing/

/usr/local/lib64/python3.6/site-packages/pandas/tests/indexing

Name	Size	Mode	Actions
interval/	-	0755	rm
multiindex/	-	0755	rm
__pycache__/	-	0755	rm
common.py	5264	0644	edit dl rm
test_callable.py	8529	0644	edit dl rm
test_categorical.py	28955	0644	edit dl rm
test_chaining_and_caching.py	13074	0644	edit dl rm
test_check_indexer.py	2917	0644	edit dl rm
test_coercion.py	38188	0644	edit dl rm
test_datetime.py	12717	0644	edit dl rm
test_floats.py	24435	0644	edit dl rm
test_iloc.py	25549	0644	edit dl rm
test_indexers.py	1517	0644	edit dl rm
test_indexing.py	37471	0644	edit dl rm
test_indexing_slow.py	418	0644	edit dl rm
test_loc.py	36948	0644	edit dl rm
test_na_indexing.py	2817	0644	edit dl rm
test_partial.py	24715	0644	edit dl rm
test_scalar.py	13259	0644	edit dl rm
test_timedelta.py	4128	0644	edit dl rm
__init__.py	0	0644	edit dl rm

Edit: /usr/local/lib64/python3.6/site-packages/pandas/tests/indexing/test_na_indexing.py (2817B)

import pytest

import pandas as pd
import pandas._testing as tm

@pytest.mark.parametrize(
    "values, dtype",
    [
        ([], "object"),
        ([1, 2, 3], "int64"),
        ([1.0, 2.0, 3.0], "float64"),
        (["a", "b", "c"], "object"),
        (["a", "b", "c"], "string"),
        ([1, 2, 3], "datetime64[ns]"),
        ([1, 2, 3], "datetime64[ns, CET]"),
        ([1, 2, 3], "timedelta64[ns]"),
        (["2000", "2001", "2002"], "Period[D]"),
        ([1, 0, 3], "Sparse"),
        ([pd.Interval(0, 1), pd.Interval(1, 2), pd.Interval(3, 4)], "interval"),
    ],
)
@pytest.mark.parametrize(
    "mask", [[True, False, False], [True, True, True], [False, False, False]]
)
@pytest.mark.parametrize("indexer_class", [list, pd.array, pd.Index, pd.Series])
@pytest.mark.parametrize("frame", [True, False])
def test_series_mask_boolean(values, dtype, mask, indexer_class, frame):
    # In case len(values) < 3
    index = ["a", "b", "c"][: len(values)]
    mask = mask[: len(values)]

obj = pd.Series(values, dtype=dtype, index=index)
    if frame:
        if len(values) == 0:
            # Otherwise obj is an empty DataFrame with shape (0, 1)
            obj = pd.DataFrame(dtype=dtype)
        else:
            obj = obj.to_frame()

if indexer_class is pd.array:
        mask = pd.array(mask, dtype="boolean")
    elif indexer_class is pd.Series:
        mask = pd.Series(mask, index=obj.index, dtype="boolean")
    else:
        mask = indexer_class(mask)

expected = obj[mask]

result = obj[mask]
    tm.assert_equal(result, expected)

if indexer_class is pd.Series:
        msg = "iLocation based boolean indexing cannot use an indexable as a mask"
        with pytest.raises(ValueError, match=msg):
            result = obj.iloc[mask]
            tm.assert_equal(result, expected)
    else:
        result = obj.iloc[mask]
        tm.assert_equal(result, expected)

result = obj.loc[mask]
    tm.assert_equal(result, expected)

@pytest.mark.parametrize("frame", [True, False])
def test_na_treated_as_false(frame):
    # https://github.com/pandas-dev/pandas/issues/31503
    s = pd.Series([1, 2, 3], name="name")

if frame:
        s = s.to_frame()

mask = pd.array([True, False, None], dtype="boolean")

result = s[mask]
    expected = s[mask.fillna(False)]

result_loc = s.loc[mask]
    expected_loc = s.loc[mask.fillna(False)]

result_iloc = s.iloc[mask]
    expected_iloc = s.iloc[mask.fillna(False)]

if frame:
        tm.assert_frame_equal(result, expected)
        tm.assert_frame_equal(result_loc, expected_loc)
        tm.assert_frame_equal(result_iloc, expected_iloc)
    else:
        tm.assert_series_equal(result, expected)
        tm.assert_series_equal(result_loc, expected_loc)
        tm.assert_series_equal(result_iloc, expected_iloc)

cmd: