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_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_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_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 assert_sp_list_equal(left, right): assert_sp_array_equal(left.to_array(), right.to_array())