/usr/share/pyshared/pandas/sparse/tests/test_list.py is in python-pandas 0.7.0-1.
This file is owned by root:root, with mode 0o644.
The actual contents of the file can be viewed below.
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 | import unittest
from numpy import nan
import numpy as np
from pandas.sparse.api import SparseList, SparseArray
from pandas.util.testing import assert_almost_equal
from test_sparse import assert_sp_array_equal
def assert_sp_list_equal(left, right):
assert_sp_array_equal(left.to_array(), right.to_array())
class TestSparseList(unittest.TestCase):
def setUp(self):
self.na_data = np.array([nan, nan, 1, 2, 3, nan, 4, 5, nan, 6])
self.zero_data = np.array([0, 0, 1, 2, 3, 0, 4, 5, 0, 6])
def test_constructor(self):
lst1 = SparseList(self.na_data[:5])
exp = SparseList()
exp.append(self.na_data[:5])
assert_sp_list_equal(lst1, exp)
def test_len(self):
arr = self.na_data
splist = SparseList()
splist.append(arr[:5])
self.assertEquals(len(splist), 5)
splist.append(arr[5])
self.assertEquals(len(splist), 6)
splist.append(arr[6:])
self.assertEquals(len(splist), 10)
def test_append_na(self):
arr = self.na_data
splist = SparseList()
splist.append(arr[:5])
splist.append(arr[5])
splist.append(arr[6:])
sparr = splist.to_array()
assert_sp_array_equal(sparr, SparseArray(arr))
def test_append_zero(self):
arr = self.zero_data
splist = SparseList(fill_value=0)
splist.append(arr[:5])
splist.append(arr[5])
splist.append(arr[6:])
sparr = splist.to_array()
assert_sp_array_equal(sparr, SparseArray(arr, fill_value=0))
def test_consolidate(self):
arr = self.na_data
exp_sparr = SparseArray(arr)
splist = SparseList()
splist.append(arr[:5])
splist.append(arr[5])
splist.append(arr[6:])
consol = splist.consolidate(inplace=False)
self.assert_(consol.nchunks == 1)
self.assert_(splist.nchunks == 3)
assert_sp_array_equal(consol.to_array(), exp_sparr)
splist.consolidate()
self.assert_(splist.nchunks == 1)
assert_sp_array_equal(splist.to_array(), exp_sparr)
def test_copy(self):
arr = self.na_data
exp_sparr = SparseArray(arr)
splist = SparseList()
splist.append(arr[:5])
splist.append(arr[5])
cp = splist.copy()
cp.append(arr[6:])
self.assertEquals(splist.nchunks, 2)
assert_sp_array_equal(cp.to_array(), exp_sparr)
def test_getitem(self):
arr = self.na_data
splist = SparseList()
splist.append(arr[:5])
splist.append(arr[5])
splist.append(arr[6:])
for i in range(len(arr)):
assert_almost_equal(splist[i], arr[i])
assert_almost_equal(splist[-i], arr[-i])
if __name__ == '__main__':
import nose
nose.runmodule(argv=[__file__,'-vvs','-x','--pdb', '--pdb-failure'],
exit=False)
|