FM / multiindex

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

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

Name	Size	Mode	Actions
__pycache__/	-	0755	rm
conftest.py	534	0644	edit dl rm
test_chaining_and_caching.py	1979	0644	edit dl rm
test_datetime.py	1209	0644	edit dl rm
test_getitem.py	8767	0644	edit dl rm
test_iloc.py	4991	0644	edit dl rm
test_indexing_slow.py	3119	0644	edit dl rm
test_insert.py	1023	0644	edit dl rm
test_ix.py	2117	0644	edit dl rm
test_loc.py	17852	0644	edit dl rm
test_multiindex.py	3263	0644	edit dl rm
test_partial.py	7614	0644	edit dl rm
test_setitem.py	15792	0644	edit dl rm
test_slice.py	25339	0644	edit dl rm
test_sorted.py	4447	0644	edit dl rm
test_xs.py	8411	0644	edit dl rm
__init__.py	0	0644	edit dl rm

Edit: /usr/local/lib64/python3.6/site-packages/pandas/tests/indexing/multiindex/test_insert.py (1023B)

from numpy.random import randn

from pandas import DataFrame, MultiIndex, Series
import pandas._testing as tm

class TestMultiIndexInsertion:
    def test_setitem_mixed_depth(self):
        arrays = [
            ["a", "top", "top", "routine1", "routine1", "routine2"],
            ["", "OD", "OD", "result1", "result2", "result1"],
            ["", "wx", "wy", "", "", ""],
        ]

tuples = sorted(zip(*arrays))
        index = MultiIndex.from_tuples(tuples)
        df = DataFrame(randn(4, 6), columns=index)

result = df.copy()
        expected = df.copy()
        result["b"] = [1, 2, 3, 4]
        expected["b", "", ""] = [1, 2, 3, 4]
        tm.assert_frame_equal(result, expected)

def test_dataframe_insert_column_all_na(self):
        # GH #1534
        mix = MultiIndex.from_tuples([("1a", "2a"), ("1a", "2b"), ("1a", "2c")])
        df = DataFrame([[1, 2], [3, 4], [5, 6]], index=mix)
        s = Series({(1, 1): 1, (1, 2): 2})
        df["new"] = s
        assert df["new"].isna().all()

from numpy.random import randn from pandas import DataFrame, MultiIndex, Series import pandas._testing as tm class TestMultiIndexInsertion: def test_setitem_mixed_depth(self): arrays = [ ["a", "top", "top", "routine1", "routine1", "routine2"], ["", "OD", "OD", "result1", "result2", "result1"], ["", "wx", "wy", "", "", ""], ] tuples = sorted(zip(*arrays)) index = MultiIndex.from_tuples(tuples) df = DataFrame(randn(4, 6), columns=index) result = df.copy() expected = df.copy() result["b"] = [1, 2, 3, 4] expected["b", "", ""] = [1, 2, 3, 4] tm.assert_frame_equal(result, expected) def test_dataframe_insert_column_all_na(self): # GH #1534 mix = MultiIndex.from_tuples([("1a", "2a"), ("1a", "2b"), ("1a", "2c")]) df = DataFrame([[1, 2], [3, 4], [5, 6]], index=mix) s = Series({(1, 1): 1, (1, 2): 2}) df["new"] = s assert df["new"].isna().all()