Python pandas.util.testing.SubclassedSparseSeries() Examples
The following are 20
code examples of pandas.util.testing.SubclassedSparseSeries().
You can vote up the ones you like or vote down the ones you don't like,
and go to the original project or source file by following the links above each example.
You may also want to check out all available functions/classes of the module
pandas.util.testing
, or try the search function
.
.
Example #1
| Source File: test_subclass.py From predictive-maintenance-using-machine-learning with Apache License 2.0 | 6 votes |
|
def test_subclass_sparse_to_frame(self):
s = tm.SubclassedSparseSeries([1, 2], index=list('ab'), name='xxx')
res = s.to_frame()
exp_arr = pd.SparseArray([1, 2], dtype=np.int64, kind='block',
fill_value=0)
exp = tm.SubclassedSparseDataFrame({'xxx': exp_arr},
index=list('ab'),
default_fill_value=0)
tm.assert_sp_frame_equal(res, exp)
# create from int dict
res = tm.SubclassedSparseDataFrame({'xxx': [1, 2]},
index=list('ab'),
default_fill_value=0)
tm.assert_sp_frame_equal(res, exp)
s = tm.SubclassedSparseSeries([1.1, 2.1], index=list('ab'),
name='xxx')
res = s.to_frame()
exp = tm.SubclassedSparseDataFrame({'xxx': [1.1, 2.1]},
index=list('ab'))
tm.assert_sp_frame_equal(res, exp)
Example #2
| Source File: test_subclass.py From twitter-stock-recommendation with MIT License | 6 votes |
|
def test_subclass_sparse_slice(self):
rows = [[1.0, 2.0, 3.0], [4.0, 5.0, 6.0], [7.0, 8.0, 9.0]]
ssdf = tm.SubclassedSparseDataFrame(rows)
ssdf.testattr = "testattr"
tm.assert_sp_frame_equal(ssdf.loc[:2],
tm.SubclassedSparseDataFrame(rows[:3]))
tm.assert_sp_frame_equal(ssdf.iloc[:2],
tm.SubclassedSparseDataFrame(rows[:2]))
tm.assert_sp_frame_equal(ssdf[:2],
tm.SubclassedSparseDataFrame(rows[:2]))
assert ssdf.loc[:2].testattr == "testattr"
assert ssdf.iloc[:2].testattr == "testattr"
assert ssdf[:2].testattr == "testattr"
tm.assert_sp_series_equal(ssdf.loc[1],
tm.SubclassedSparseSeries(rows[1]),
check_names=False)
tm.assert_sp_series_equal(ssdf.iloc[1],
tm.SubclassedSparseSeries(rows[1]),
check_names=False)
Example #3
| Source File: test_subclass.py From twitter-stock-recommendation with MIT License | 6 votes |
|
def test_subclass_sparse_to_frame(self):
s = tm.SubclassedSparseSeries([1, 2], index=list('abcd'), name='xxx')
res = s.to_frame()
exp_arr = pd.SparseArray([1, 2], dtype=np.int64, kind='block',
fill_value=0)
exp = tm.SubclassedSparseDataFrame({'xxx': exp_arr},
index=list('abcd'),
default_fill_value=0)
tm.assert_sp_frame_equal(res, exp)
# create from int dict
res = tm.SubclassedSparseDataFrame({'xxx': [1, 2]},
index=list('abcd'),
default_fill_value=0)
tm.assert_sp_frame_equal(res, exp)
s = tm.SubclassedSparseSeries([1.1, 2.1], index=list('abcd'),
name='xxx')
res = s.to_frame()
exp = tm.SubclassedSparseDataFrame({'xxx': [1.1, 2.1]},
index=list('abcd'))
tm.assert_sp_frame_equal(res, exp)
Example #4
| Source File: test_subclass.py From elasticintel with GNU General Public License v3.0 | 6 votes |
|
def test_subclass_sparse_slice(self):
rows = [[1.0, 2.0, 3.0], [4.0, 5.0, 6.0], [7.0, 8.0, 9.0]]
ssdf = tm.SubclassedSparseDataFrame(rows)
ssdf.testattr = "testattr"
tm.assert_sp_frame_equal(ssdf.loc[:2],
tm.SubclassedSparseDataFrame(rows[:3]))
tm.assert_sp_frame_equal(ssdf.iloc[:2],
tm.SubclassedSparseDataFrame(rows[:2]))
tm.assert_sp_frame_equal(ssdf[:2],
tm.SubclassedSparseDataFrame(rows[:2]))
assert ssdf.loc[:2].testattr == "testattr"
assert ssdf.iloc[:2].testattr == "testattr"
assert ssdf[:2].testattr == "testattr"
tm.assert_sp_series_equal(ssdf.loc[1],
tm.SubclassedSparseSeries(rows[1]),
check_names=False)
tm.assert_sp_series_equal(ssdf.iloc[1],
tm.SubclassedSparseSeries(rows[1]),
check_names=False)
Example #5
| Source File: test_subclass.py From elasticintel with GNU General Public License v3.0 | 6 votes |
|
def test_subclass_sparse_to_frame(self):
s = tm.SubclassedSparseSeries([1, 2], index=list('abcd'), name='xxx')
res = s.to_frame()
exp_arr = pd.SparseArray([1, 2], dtype=np.int64, kind='block',
fill_value=0)
exp = tm.SubclassedSparseDataFrame({'xxx': exp_arr},
index=list('abcd'),
default_fill_value=0)
tm.assert_sp_frame_equal(res, exp)
# create from int dict
res = tm.SubclassedSparseDataFrame({'xxx': [1, 2]},
index=list('abcd'),
default_fill_value=0)
tm.assert_sp_frame_equal(res, exp)
s = tm.SubclassedSparseSeries([1.1, 2.1], index=list('abcd'),
name='xxx')
res = s.to_frame()
exp = tm.SubclassedSparseDataFrame({'xxx': [1.1, 2.1]},
index=list('abcd'))
tm.assert_sp_frame_equal(res, exp)
Example #6
| Source File: test_subclass.py From predictive-maintenance-using-machine-learning with Apache License 2.0 | 6 votes |
|
def test_subclass_sparse_slice(self):
rows = [[1.0, 2.0, 3.0], [4.0, 5.0, 6.0], [7.0, 8.0, 9.0]]
ssdf = tm.SubclassedSparseDataFrame(rows)
ssdf.testattr = "testattr"
tm.assert_sp_frame_equal(ssdf.loc[:2],
tm.SubclassedSparseDataFrame(rows[:3]))
tm.assert_sp_frame_equal(ssdf.iloc[:2],
tm.SubclassedSparseDataFrame(rows[:2]))
tm.assert_sp_frame_equal(ssdf[:2],
tm.SubclassedSparseDataFrame(rows[:2]))
assert ssdf.loc[:2].testattr == "testattr"
assert ssdf.iloc[:2].testattr == "testattr"
assert ssdf[:2].testattr == "testattr"
tm.assert_sp_series_equal(ssdf.loc[1],
tm.SubclassedSparseSeries(rows[1]),
check_names=False,
check_kind=False)
tm.assert_sp_series_equal(ssdf.iloc[1],
tm.SubclassedSparseSeries(rows[1]),
check_names=False,
check_kind=False)
Example #7
| Source File: test_subclass.py From vnpy_crypto with MIT License | 6 votes |
|
def test_subclass_sparse_slice(self):
rows = [[1.0, 2.0, 3.0], [4.0, 5.0, 6.0], [7.0, 8.0, 9.0]]
ssdf = tm.SubclassedSparseDataFrame(rows)
ssdf.testattr = "testattr"
tm.assert_sp_frame_equal(ssdf.loc[:2],
tm.SubclassedSparseDataFrame(rows[:3]))
tm.assert_sp_frame_equal(ssdf.iloc[:2],
tm.SubclassedSparseDataFrame(rows[:2]))
tm.assert_sp_frame_equal(ssdf[:2],
tm.SubclassedSparseDataFrame(rows[:2]))
assert ssdf.loc[:2].testattr == "testattr"
assert ssdf.iloc[:2].testattr == "testattr"
assert ssdf[:2].testattr == "testattr"
tm.assert_sp_series_equal(ssdf.loc[1],
tm.SubclassedSparseSeries(rows[1]),
check_names=False)
tm.assert_sp_series_equal(ssdf.iloc[1],
tm.SubclassedSparseSeries(rows[1]),
check_names=False)
Example #8
| Source File: test_subclass.py From vnpy_crypto with MIT License | 6 votes |
|
def test_subclass_sparse_to_frame(self):
s = tm.SubclassedSparseSeries([1, 2], index=list('abcd'), name='xxx')
res = s.to_frame()
exp_arr = pd.SparseArray([1, 2], dtype=np.int64, kind='block',
fill_value=0)
exp = tm.SubclassedSparseDataFrame({'xxx': exp_arr},
index=list('abcd'),
default_fill_value=0)
tm.assert_sp_frame_equal(res, exp)
# create from int dict
res = tm.SubclassedSparseDataFrame({'xxx': [1, 2]},
index=list('abcd'),
default_fill_value=0)
tm.assert_sp_frame_equal(res, exp)
s = tm.SubclassedSparseSeries([1.1, 2.1], index=list('abcd'),
name='xxx')
res = s.to_frame()
exp = tm.SubclassedSparseDataFrame({'xxx': [1.1, 2.1]},
index=list('abcd'))
tm.assert_sp_frame_equal(res, exp)
Example #9
| Source File: test_subclass.py From recruit with Apache License 2.0 | 6 votes |
|
def test_subclass_sparse_slice(self):
rows = [[1.0, 2.0, 3.0], [4.0, 5.0, 6.0], [7.0, 8.0, 9.0]]
ssdf = tm.SubclassedSparseDataFrame(rows)
ssdf.testattr = "testattr"
tm.assert_sp_frame_equal(ssdf.loc[:2],
tm.SubclassedSparseDataFrame(rows[:3]))
tm.assert_sp_frame_equal(ssdf.iloc[:2],
tm.SubclassedSparseDataFrame(rows[:2]))
tm.assert_sp_frame_equal(ssdf[:2],
tm.SubclassedSparseDataFrame(rows[:2]))
assert ssdf.loc[:2].testattr == "testattr"
assert ssdf.iloc[:2].testattr == "testattr"
assert ssdf[:2].testattr == "testattr"
tm.assert_sp_series_equal(ssdf.loc[1],
tm.SubclassedSparseSeries(rows[1]),
check_names=False,
check_kind=False)
tm.assert_sp_series_equal(ssdf.iloc[1],
tm.SubclassedSparseSeries(rows[1]),
check_names=False,
check_kind=False)
Example #10
| Source File: test_subclass.py From recruit with Apache License 2.0 | 6 votes |
|
def test_subclass_sparse_to_frame(self):
s = tm.SubclassedSparseSeries([1, 2], index=list('ab'), name='xxx')
res = s.to_frame()
exp_arr = pd.SparseArray([1, 2], dtype=np.int64, kind='block',
fill_value=0)
exp = tm.SubclassedSparseDataFrame({'xxx': exp_arr},
index=list('ab'),
default_fill_value=0)
tm.assert_sp_frame_equal(res, exp)
# create from int dict
res = tm.SubclassedSparseDataFrame({'xxx': [1, 2]},
index=list('ab'),
default_fill_value=0)
tm.assert_sp_frame_equal(res, exp)
s = tm.SubclassedSparseSeries([1.1, 2.1], index=list('ab'),
name='xxx')
res = s.to_frame()
exp = tm.SubclassedSparseDataFrame({'xxx': [1.1, 2.1]},
index=list('ab'))
tm.assert_sp_frame_equal(res, exp)
Example #11
| Source File: test_subclass.py From predictive-maintenance-using-machine-learning with Apache License 2.0 | 5 votes |
|
def test_subclass_sparse_slice(self):
# int64
s = tm.SubclassedSparseSeries([1, 2, 3, 4, 5])
exp = tm.SubclassedSparseSeries([2, 3, 4], index=[1, 2, 3])
tm.assert_sp_series_equal(s.loc[1:3], exp)
assert s.loc[1:3].dtype == SparseDtype(np.int64)
exp = tm.SubclassedSparseSeries([2, 3], index=[1, 2])
tm.assert_sp_series_equal(s.iloc[1:3], exp)
assert s.iloc[1:3].dtype == SparseDtype(np.int64)
exp = tm.SubclassedSparseSeries([2, 3], index=[1, 2])
tm.assert_sp_series_equal(s[1:3], exp)
assert s[1:3].dtype == SparseDtype(np.int64)
# float64
s = tm.SubclassedSparseSeries([1., 2., 3., 4., 5.])
exp = tm.SubclassedSparseSeries([2., 3., 4.], index=[1, 2, 3])
tm.assert_sp_series_equal(s.loc[1:3], exp)
assert s.loc[1:3].dtype == SparseDtype(np.float64)
exp = tm.SubclassedSparseSeries([2., 3.], index=[1, 2])
tm.assert_sp_series_equal(s.iloc[1:3], exp)
assert s.iloc[1:3].dtype == SparseDtype(np.float64)
exp = tm.SubclassedSparseSeries([2., 3.], index=[1, 2])
tm.assert_sp_series_equal(s[1:3], exp)
assert s[1:3].dtype == SparseDtype(np.float64)
Example #12
| Source File: test_subclass.py From predictive-maintenance-using-machine-learning with Apache License 2.0 | 5 votes |
|
def test_subclass_sparse_addition(self):
s1 = tm.SubclassedSparseSeries([1, 3, 5])
s2 = tm.SubclassedSparseSeries([-2, 5, 12])
exp = tm.SubclassedSparseSeries([-1, 8, 17])
tm.assert_sp_series_equal(s1 + s2, exp)
s1 = tm.SubclassedSparseSeries([4.0, 5.0, 6.0])
s2 = tm.SubclassedSparseSeries([1.0, 2.0, 3.0])
exp = tm.SubclassedSparseSeries([5., 7., 9.])
tm.assert_sp_series_equal(s1 + s2, exp)
Example #13
| Source File: test_subclass.py From recruit with Apache License 2.0 | 5 votes |
|
def test_subclass_sparse_slice(self):
# int64
s = tm.SubclassedSparseSeries([1, 2, 3, 4, 5])
exp = tm.SubclassedSparseSeries([2, 3, 4], index=[1, 2, 3])
tm.assert_sp_series_equal(s.loc[1:3], exp)
assert s.loc[1:3].dtype == SparseDtype(np.int64)
exp = tm.SubclassedSparseSeries([2, 3], index=[1, 2])
tm.assert_sp_series_equal(s.iloc[1:3], exp)
assert s.iloc[1:3].dtype == SparseDtype(np.int64)
exp = tm.SubclassedSparseSeries([2, 3], index=[1, 2])
tm.assert_sp_series_equal(s[1:3], exp)
assert s[1:3].dtype == SparseDtype(np.int64)
# float64
s = tm.SubclassedSparseSeries([1., 2., 3., 4., 5.])
exp = tm.SubclassedSparseSeries([2., 3., 4.], index=[1, 2, 3])
tm.assert_sp_series_equal(s.loc[1:3], exp)
assert s.loc[1:3].dtype == SparseDtype(np.float64)
exp = tm.SubclassedSparseSeries([2., 3.], index=[1, 2])
tm.assert_sp_series_equal(s.iloc[1:3], exp)
assert s.iloc[1:3].dtype == SparseDtype(np.float64)
exp = tm.SubclassedSparseSeries([2., 3.], index=[1, 2])
tm.assert_sp_series_equal(s[1:3], exp)
assert s[1:3].dtype == SparseDtype(np.float64)
Example #14
| Source File: test_subclass.py From vnpy_crypto with MIT License | 5 votes |
|
def test_subclass_sparse_addition(self):
s1 = tm.SubclassedSparseSeries([1, 3, 5])
s2 = tm.SubclassedSparseSeries([-2, 5, 12])
exp = tm.SubclassedSparseSeries([-1, 8, 17])
tm.assert_sp_series_equal(s1 + s2, exp)
s1 = tm.SubclassedSparseSeries([4.0, 5.0, 6.0])
s2 = tm.SubclassedSparseSeries([1.0, 2.0, 3.0])
exp = tm.SubclassedSparseSeries([5., 7., 9.])
tm.assert_sp_series_equal(s1 + s2, exp)
Example #15
| Source File: test_subclass.py From elasticintel with GNU General Public License v3.0 | 5 votes |
|
def test_subclass_sparse_slice(self):
# int64
s = tm.SubclassedSparseSeries([1, 2, 3, 4, 5])
exp = tm.SubclassedSparseSeries([2, 3, 4], index=[1, 2, 3])
tm.assert_sp_series_equal(s.loc[1:3], exp)
assert s.loc[1:3].dtype == np.int64
exp = tm.SubclassedSparseSeries([2, 3], index=[1, 2])
tm.assert_sp_series_equal(s.iloc[1:3], exp)
assert s.iloc[1:3].dtype == np.int64
exp = tm.SubclassedSparseSeries([2, 3], index=[1, 2])
tm.assert_sp_series_equal(s[1:3], exp)
assert s[1:3].dtype == np.int64
# float64
s = tm.SubclassedSparseSeries([1., 2., 3., 4., 5.])
exp = tm.SubclassedSparseSeries([2., 3., 4.], index=[1, 2, 3])
tm.assert_sp_series_equal(s.loc[1:3], exp)
assert s.loc[1:3].dtype == np.float64
exp = tm.SubclassedSparseSeries([2., 3.], index=[1, 2])
tm.assert_sp_series_equal(s.iloc[1:3], exp)
assert s.iloc[1:3].dtype == np.float64
exp = tm.SubclassedSparseSeries([2., 3.], index=[1, 2])
tm.assert_sp_series_equal(s[1:3], exp)
assert s[1:3].dtype == np.float64
Example #16
| Source File: test_subclass.py From elasticintel with GNU General Public License v3.0 | 5 votes |
|
def test_subclass_sparse_addition(self):
s1 = tm.SubclassedSparseSeries([1, 3, 5])
s2 = tm.SubclassedSparseSeries([-2, 5, 12])
exp = tm.SubclassedSparseSeries([-1, 8, 17])
tm.assert_sp_series_equal(s1 + s2, exp)
s1 = tm.SubclassedSparseSeries([4.0, 5.0, 6.0])
s2 = tm.SubclassedSparseSeries([1.0, 2.0, 3.0])
exp = tm.SubclassedSparseSeries([5., 7., 9.])
tm.assert_sp_series_equal(s1 + s2, exp)
Example #17
| Source File: test_subclass.py From vnpy_crypto with MIT License | 5 votes |
|
def test_subclass_sparse_slice(self):
# int64
s = tm.SubclassedSparseSeries([1, 2, 3, 4, 5])
exp = tm.SubclassedSparseSeries([2, 3, 4], index=[1, 2, 3])
tm.assert_sp_series_equal(s.loc[1:3], exp)
assert s.loc[1:3].dtype == np.int64
exp = tm.SubclassedSparseSeries([2, 3], index=[1, 2])
tm.assert_sp_series_equal(s.iloc[1:3], exp)
assert s.iloc[1:3].dtype == np.int64
exp = tm.SubclassedSparseSeries([2, 3], index=[1, 2])
tm.assert_sp_series_equal(s[1:3], exp)
assert s[1:3].dtype == np.int64
# float64
s = tm.SubclassedSparseSeries([1., 2., 3., 4., 5.])
exp = tm.SubclassedSparseSeries([2., 3., 4.], index=[1, 2, 3])
tm.assert_sp_series_equal(s.loc[1:3], exp)
assert s.loc[1:3].dtype == np.float64
exp = tm.SubclassedSparseSeries([2., 3.], index=[1, 2])
tm.assert_sp_series_equal(s.iloc[1:3], exp)
assert s.iloc[1:3].dtype == np.float64
exp = tm.SubclassedSparseSeries([2., 3.], index=[1, 2])
tm.assert_sp_series_equal(s[1:3], exp)
assert s[1:3].dtype == np.float64
Example #18
| Source File: test_subclass.py From twitter-stock-recommendation with MIT License | 5 votes |
|
def test_subclass_sparse_slice(self):
# int64
s = tm.SubclassedSparseSeries([1, 2, 3, 4, 5])
exp = tm.SubclassedSparseSeries([2, 3, 4], index=[1, 2, 3])
tm.assert_sp_series_equal(s.loc[1:3], exp)
assert s.loc[1:3].dtype == np.int64
exp = tm.SubclassedSparseSeries([2, 3], index=[1, 2])
tm.assert_sp_series_equal(s.iloc[1:3], exp)
assert s.iloc[1:3].dtype == np.int64
exp = tm.SubclassedSparseSeries([2, 3], index=[1, 2])
tm.assert_sp_series_equal(s[1:3], exp)
assert s[1:3].dtype == np.int64
# float64
s = tm.SubclassedSparseSeries([1., 2., 3., 4., 5.])
exp = tm.SubclassedSparseSeries([2., 3., 4.], index=[1, 2, 3])
tm.assert_sp_series_equal(s.loc[1:3], exp)
assert s.loc[1:3].dtype == np.float64
exp = tm.SubclassedSparseSeries([2., 3.], index=[1, 2])
tm.assert_sp_series_equal(s.iloc[1:3], exp)
assert s.iloc[1:3].dtype == np.float64
exp = tm.SubclassedSparseSeries([2., 3.], index=[1, 2])
tm.assert_sp_series_equal(s[1:3], exp)
assert s[1:3].dtype == np.float64
Example #19
| Source File: test_subclass.py From twitter-stock-recommendation with MIT License | 5 votes |
|
def test_subclass_sparse_addition(self):
s1 = tm.SubclassedSparseSeries([1, 3, 5])
s2 = tm.SubclassedSparseSeries([-2, 5, 12])
exp = tm.SubclassedSparseSeries([-1, 8, 17])
tm.assert_sp_series_equal(s1 + s2, exp)
s1 = tm.SubclassedSparseSeries([4.0, 5.0, 6.0])
s2 = tm.SubclassedSparseSeries([1.0, 2.0, 3.0])
exp = tm.SubclassedSparseSeries([5., 7., 9.])
tm.assert_sp_series_equal(s1 + s2, exp)
Example #20
| Source File: test_subclass.py From recruit with Apache License 2.0 | 5 votes |
|
def test_subclass_sparse_addition(self):
s1 = tm.SubclassedSparseSeries([1, 3, 5])
s2 = tm.SubclassedSparseSeries([-2, 5, 12])
exp = tm.SubclassedSparseSeries([-1, 8, 17])
tm.assert_sp_series_equal(s1 + s2, exp)
s1 = tm.SubclassedSparseSeries([4.0, 5.0, 6.0])
s2 = tm.SubclassedSparseSeries([1.0, 2.0, 3.0])
exp = tm.SubclassedSparseSeries([5., 7., 9.])
tm.assert_sp_series_equal(s1 + s2, exp)
