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.assertEqual(consol.nchunks, 1) self.assertEqual(splist.nchunks, 3) assert_sp_array_equal(consol.to_array(), exp_sparr) splist.consolidate() self.assertEqual(splist.nchunks, 1) assert_sp_array_equal(splist.to_array(), exp_sparr)
def test_consolidate(self): with tm.assert_produces_warning(FutureWarning, check_stacklevel=False): 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.assertEqual(consol.nchunks, 1) self.assertEqual(splist.nchunks, 3) tm.assert_sp_array_equal(consol.to_array(), exp_sparr) splist.consolidate() self.assertEqual(splist.nchunks, 1) tm.assert_sp_array_equal(splist.to_array(), exp_sparr)