Пример #1
0
 def test_raise_when_not_implemented(self):
     p = Panel(np.arange(3 * 4 * 5).reshape(3, 4, 5),
               items=['ItemA', 'ItemB', 'ItemC'],
               major_axis=date_range('20130101', periods=4),
               minor_axis=list('ABCDE'))
     d = p.sum(axis=1).iloc[0]
     ops = ['add', 'sub', 'mul', 'truediv', 'floordiv', 'div', 'mod', 'pow']
     for op in ops:
         with pytest.raises(NotImplementedError):
             getattr(p, op)(d, axis=0)
Пример #2
0
    def test_apply_slabs(self):
        # with multi-indexes
        # GH7469
        index = MultiIndex.from_tuples([('one', 'a'), ('one', 'b'),
                                        ('two', 'a'), ('two', 'b')])
        dfa = DataFrame(np.array(np.arange(12, dtype='int64')).reshape(4, 3),
                        columns=list("ABC"),
                        index=index)
        dfb = DataFrame(np.array(np.arange(10, 22,
                                           dtype='int64')).reshape(4, 3),
                        columns=list("ABC"),
                        index=index)
        p = Panel({'f': dfa, 'g': dfb})
        result = p.apply(lambda x: x.sum(), axis=0)

        # on windows this will be in32
        result = result.astype('int64')
        expected = p.sum(0)
        assert_frame_equal(result, expected)