Example #1
0
    def setup_method(self, method):
        super(TestIndex, self).setup_method(method)

        self.d = {
            'string': tm.makeStringIndex(100),
            'date': tm.makeDateIndex(100),
            'int': tm.makeIntIndex(100),
            'rng': tm.makeRangeIndex(100),
            'float': tm.makeFloatIndex(100),
            'empty': Index([]),
            'tuple': Index(zip(['foo', 'bar', 'baz'], [1, 2, 3])),
            'period': Index(period_range('2012-1-1', freq='M', periods=3)),
            'date2': Index(date_range('2013-01-1', periods=10)),
            'bdate': Index(bdate_range('2013-01-02', periods=10)),
            'cat': tm.makeCategoricalIndex(100),
            'interval': tm.makeIntervalIndex(100),
            'timedelta': tm.makeTimedeltaIndex(100, 'H')
        }

        self.mi = {
            'reg': MultiIndex.from_tuples([('bar', 'one'), ('baz', 'two'),
                                           ('foo', 'two'),
                                           ('qux', 'one'), ('qux', 'two')],
                                          names=['first', 'second']),
        }
Example #2
0
 def setup_method(self, method):
     self.index = IntervalIndex.from_arrays([0, 1], [1, 2])
     self.index_with_nan = IntervalIndex.from_tuples(
         [(0, 1), np.nan, (1, 2)])
     self.indices = dict(intervalIndex=tm.makeIntervalIndex(10))
Example #3
0
 def setup_method(self, method):
     self.index = IntervalIndex.from_arrays([0, 1], [1, 2])
     self.index_with_nan = IntervalIndex.from_tuples([(0, 1), np.nan,
                                                      (1, 2)])
     self.indices = dict(intervalIndex=tm.makeIntervalIndex(10))
Example #4
0
import pandas as pd
from pandas.core.indexes.api import Index, MultiIndex
import pandas.util.testing as tm

indices_list = [tm.makeUnicodeIndex(100),
                tm.makeStringIndex(100),
                tm.makeDateIndex(100),
                tm.makePeriodIndex(100),
                tm.makeTimedeltaIndex(100),
                tm.makeIntIndex(100),
                tm.makeUIntIndex(100),
                tm.makeRangeIndex(100),
                tm.makeFloatIndex(100),
                Index([True, False]),
                tm.makeCategoricalIndex(100),
                tm.makeIntervalIndex(100),
                Index([]),
                MultiIndex.from_tuples(zip(
                    ['foo', 'bar', 'baz'], [1, 2, 3])),
                Index([0, 0, 1, 1, 2, 2])]


@pytest.fixture(params=indices_list, ids=lambda x: type(x).__name__)
def indices(request):
    return request.param


@pytest.fixture(params=[1, np.array(1, dtype=np.int64)])
def one(request):
    # zero-dim integer array behaves like an integer
    return request.param