def setUp(self):
major_axis = Index([1, 2, 3, 4])
minor_axis = Index([1, 2])
major_labels = np.array([0, 0, 1, 2, 3, 3])
minor_labels = np.array([0, 1, 0, 1, 0, 1])
self.index = LongPanelIndex(major_axis, minor_axis,
major_labels, minor_labels)
major_labels = np.array([0, 0, 1, 1, 1, 2, 2, 3, 3])
minor_labels = np.array([0, 1, 0, 1, 1, 0, 1, 0, 1])
self.incon = LongPanelIndex(major_axis, minor_axis,
major_labels, minor_labels)
def test_consistency(self):
self.assert_(self.index.consistent)
self.assert_(not self.incon.consistent)
# need to construct an overflow
major_axis = range(70000)
minor_axis = range(10)
major_labels = np.arange(70000)
minor_labels = np.repeat(range(10), 7000)
index = LongPanelIndex(major_axis, minor_axis,
major_labels, minor_labels)
self.assert_(index.consistent)
class TestLongPanelIndex(unittest.TestCase):
def setUp(self):
major_axis = Index([1, 2, 3, 4])
minor_axis = Index([1, 2])
major_labels = np.array([0, 0, 1, 2, 3, 3])
minor_labels = np.array([0, 1, 0, 1, 0, 1])
self.index = LongPanelIndex(major_axis, minor_axis,
major_labels, minor_labels)
major_labels = np.array([0, 0, 1, 1, 1, 2, 2, 3, 3])
minor_labels = np.array([0, 1, 0, 1, 1, 0, 1, 0, 1])
self.incon = LongPanelIndex(major_axis, minor_axis,
major_labels, minor_labels)
def test_consistency(self):
self.assert_(self.index.consistent)
self.assert_(not self.incon.consistent)
# need to construct an overflow
major_axis = range(70000)
minor_axis = range(10)
major_labels = np.arange(70000)
minor_labels = np.repeat(range(10), 7000)
index = LongPanelIndex(major_axis, minor_axis,
major_labels, minor_labels)
self.assert_(index.consistent)
def test_truncate(self):
result = self.index.truncate(before=1)
self.assert_(0 not in result.major_axis)
self.assert_(1 in result.major_axis)
result = self.index.truncate(after=1)
self.assert_(2 not in result.major_axis)
self.assert_(1 in result.major_axis)
result = self.index.truncate(before=1, after=2)
self.assertEqual(len(result.major_axis), 2)
def test_getMajorBounds(self):
pass
def test_getAxisBounds(self):
pass
def test_getLabelBounds(self):
pass
def test_bounds(self):
pass
def test_makeMask(self):
mask = self.index.mask
expected = np.array([True, True,
True, False,
False, True,
True, True], dtype=bool)
self.assert_(np.array_equal(mask, expected))
def test_dims(self):
pass
