| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596 |
- # -*- coding: utf-8 -*-
- import numpy as np
- import pytest
- import pandas as pd
- from pandas import Index, MultiIndex
- import pandas.util.testing as tm
- @pytest.mark.parametrize('other', [
- Index(['three', 'one', 'two']),
- Index(['one']),
- Index(['one', 'three']),
- ])
- def test_join_level(idx, other, join_type):
- join_index, lidx, ridx = other.join(idx, how=join_type,
- level='second',
- return_indexers=True)
- exp_level = other.join(idx.levels[1], how=join_type)
- assert join_index.levels[0].equals(idx.levels[0])
- assert join_index.levels[1].equals(exp_level)
- # pare down levels
- mask = np.array(
- [x[1] in exp_level for x in idx], dtype=bool)
- exp_values = idx.values[mask]
- tm.assert_numpy_array_equal(join_index.values, exp_values)
- if join_type in ('outer', 'inner'):
- join_index2, ridx2, lidx2 = \
- idx.join(other, how=join_type, level='second',
- return_indexers=True)
- assert join_index.equals(join_index2)
- tm.assert_numpy_array_equal(lidx, lidx2)
- tm.assert_numpy_array_equal(ridx, ridx2)
- tm.assert_numpy_array_equal(join_index2.values, exp_values)
- def test_join_level_corner_case(idx):
- # some corner cases
- index = Index(['three', 'one', 'two'])
- result = index.join(idx, level='second')
- assert isinstance(result, MultiIndex)
- with pytest.raises(TypeError, match="Join.*MultiIndex.*ambiguous"):
- idx.join(idx, level=1)
- def test_join_self(idx, join_type):
- joined = idx.join(idx, how=join_type)
- assert idx is joined
- def test_join_multi():
- # GH 10665
- midx = pd.MultiIndex.from_product(
- [np.arange(4), np.arange(4)], names=['a', 'b'])
- idx = pd.Index([1, 2, 5], name='b')
- # inner
- jidx, lidx, ridx = midx.join(idx, how='inner', return_indexers=True)
- exp_idx = pd.MultiIndex.from_product(
- [np.arange(4), [1, 2]], names=['a', 'b'])
- exp_lidx = np.array([1, 2, 5, 6, 9, 10, 13, 14], dtype=np.intp)
- exp_ridx = np.array([0, 1, 0, 1, 0, 1, 0, 1], dtype=np.intp)
- tm.assert_index_equal(jidx, exp_idx)
- tm.assert_numpy_array_equal(lidx, exp_lidx)
- tm.assert_numpy_array_equal(ridx, exp_ridx)
- # flip
- jidx, ridx, lidx = idx.join(midx, how='inner', return_indexers=True)
- tm.assert_index_equal(jidx, exp_idx)
- tm.assert_numpy_array_equal(lidx, exp_lidx)
- tm.assert_numpy_array_equal(ridx, exp_ridx)
- # keep MultiIndex
- jidx, lidx, ridx = midx.join(idx, how='left', return_indexers=True)
- exp_ridx = np.array([-1, 0, 1, -1, -1, 0, 1, -1, -1, 0, 1, -1, -1, 0,
- 1, -1], dtype=np.intp)
- tm.assert_index_equal(jidx, midx)
- assert lidx is None
- tm.assert_numpy_array_equal(ridx, exp_ridx)
- # flip
- jidx, ridx, lidx = idx.join(midx, how='right', return_indexers=True)
- tm.assert_index_equal(jidx, midx)
- assert lidx is None
- tm.assert_numpy_array_equal(ridx, exp_ridx)
- def test_join_self_unique(idx, join_type):
- if idx.is_unique:
- joined = idx.join(idx, how=join_type)
- assert (idx == joined).all()
|
