@@ -16,83 +16,99 @@ def setUp(self):
16
16
self .na_data = np .array ([nan , nan , 1 , 2 , 3 , nan , 4 , 5 , nan , 6 ])
17
17
self .zero_data = np .array ([0 , 0 , 1 , 2 , 3 , 0 , 4 , 5 , 0 , 6 ])
18
18
19
+ def test_deprecation (self ):
20
+ # see gh-13784
21
+ with tm .assert_produces_warning (FutureWarning ):
22
+ SparseList ()
23
+
19
24
def test_constructor (self ):
20
- lst1 = SparseList (self .na_data [:5 ])
21
- exp = SparseList ()
25
+ with tm .assert_produces_warning (FutureWarning ):
26
+ lst1 = SparseList (self .na_data [:5 ])
27
+ with tm .assert_produces_warning (FutureWarning ):
28
+ exp = SparseList ()
29
+
22
30
exp .append (self .na_data [:5 ])
23
31
tm .assert_sp_list_equal (lst1 , exp )
24
32
25
33
def test_len (self ):
26
- arr = self .na_data
27
- splist = SparseList ()
28
- splist .append (arr [:5 ])
29
- self .assertEqual (len (splist ), 5 )
30
- splist .append (arr [5 ])
31
- self .assertEqual (len (splist ), 6 )
32
- splist .append (arr [6 :])
33
- self .assertEqual (len (splist ), 10 )
34
+ with tm .assert_produces_warning (FutureWarning ):
35
+ arr = self .na_data
36
+ splist = SparseList ()
37
+ splist .append (arr [:5 ])
38
+ self .assertEqual (len (splist ), 5 )
39
+ splist .append (arr [5 ])
40
+ self .assertEqual (len (splist ), 6 )
41
+ splist .append (arr [6 :])
42
+ self .assertEqual (len (splist ), 10 )
34
43
35
44
def test_append_na (self ):
36
- arr = self .na_data
37
- splist = SparseList ()
38
- splist .append (arr [:5 ])
39
- splist .append (arr [5 ])
40
- splist .append (arr [6 :])
45
+ with tm .assert_produces_warning (FutureWarning ):
46
+ arr = self .na_data
47
+ splist = SparseList ()
48
+ splist .append (arr [:5 ])
49
+ splist .append (arr [5 ])
50
+ splist .append (arr [6 :])
41
51
42
- sparr = splist .to_array ()
43
- tm .assert_sp_array_equal (sparr , SparseArray (arr ))
52
+ sparr = splist .to_array ()
53
+ tm .assert_sp_array_equal (sparr , SparseArray (arr ))
44
54
45
55
def test_append_zero (self ):
46
- arr = self .zero_data
47
- splist = SparseList (fill_value = 0 )
48
- splist .append (arr [:5 ])
49
- splist .append (arr [5 ])
50
- splist .append (arr [6 :])
56
+ with tm .assert_produces_warning (FutureWarning ):
57
+ arr = self .zero_data
58
+ splist = SparseList (fill_value = 0 )
59
+ splist .append (arr [:5 ])
60
+ splist .append (arr [5 ])
61
+ splist .append (arr [6 :])
51
62
52
- sparr = splist .to_array ()
53
- tm .assert_sp_array_equal (sparr , SparseArray (arr , fill_value = 0 ))
63
+ sparr = splist .to_array ()
64
+ tm .assert_sp_array_equal (sparr , SparseArray (arr , fill_value = 0 ))
54
65
55
66
def test_consolidate (self ):
56
- arr = self .na_data
57
- exp_sparr = SparseArray (arr )
67
+ with tm .assert_produces_warning (FutureWarning ,
68
+ check_stacklevel = False ):
69
+ arr = self .na_data
70
+ exp_sparr = SparseArray (arr )
58
71
59
- splist = SparseList ()
60
- splist .append (arr [:5 ])
61
- splist .append (arr [5 ])
62
- splist .append (arr [6 :])
72
+ splist = SparseList ()
73
+ splist .append (arr [:5 ])
74
+ splist .append (arr [5 ])
75
+ splist .append (arr [6 :])
63
76
64
- consol = splist .consolidate (inplace = False )
65
- self .assertEqual (consol .nchunks , 1 )
66
- self .assertEqual (splist .nchunks , 3 )
67
- tm .assert_sp_array_equal (consol .to_array (), exp_sparr )
77
+ consol = splist .consolidate (inplace = False )
78
+ self .assertEqual (consol .nchunks , 1 )
79
+ self .assertEqual (splist .nchunks , 3 )
80
+ tm .assert_sp_array_equal (consol .to_array (), exp_sparr )
68
81
69
- splist .consolidate ()
70
- self .assertEqual (splist .nchunks , 1 )
71
- tm .assert_sp_array_equal (splist .to_array (), exp_sparr )
82
+ splist .consolidate ()
83
+ self .assertEqual (splist .nchunks , 1 )
84
+ tm .assert_sp_array_equal (splist .to_array (), exp_sparr )
72
85
73
86
def test_copy (self ):
74
- arr = self .na_data
75
- exp_sparr = SparseArray (arr )
87
+ with tm .assert_produces_warning (FutureWarning ,
88
+ check_stacklevel = False ):
89
+ arr = self .na_data
90
+ exp_sparr = SparseArray (arr )
76
91
77
- splist = SparseList ()
78
- splist .append (arr [:5 ])
79
- splist .append (arr [5 ])
92
+ splist = SparseList ()
93
+ splist .append (arr [:5 ])
94
+ splist .append (arr [5 ])
80
95
81
- cp = splist .copy ()
82
- cp .append (arr [6 :])
83
- self .assertEqual (splist .nchunks , 2 )
84
- tm .assert_sp_array_equal (cp .to_array (), exp_sparr )
96
+ cp = splist .copy ()
97
+ cp .append (arr [6 :])
98
+ self .assertEqual (splist .nchunks , 2 )
99
+ tm .assert_sp_array_equal (cp .to_array (), exp_sparr )
85
100
86
101
def test_getitem (self ):
87
- arr = self .na_data
88
- splist = SparseList ()
89
- splist .append (arr [:5 ])
90
- splist .append (arr [5 ])
91
- splist .append (arr [6 :])
92
-
93
- for i in range (len (arr )):
94
- tm .assert_almost_equal (splist [i ], arr [i ])
95
- tm .assert_almost_equal (splist [- i ], arr [- i ])
102
+ with tm .assert_produces_warning (FutureWarning ):
103
+ arr = self .na_data
104
+ splist = SparseList ()
105
+ splist .append (arr [:5 ])
106
+ splist .append (arr [5 ])
107
+ splist .append (arr [6 :])
108
+
109
+ for i in range (len (arr )):
110
+ tm .assert_almost_equal (splist [i ], arr [i ])
111
+ tm .assert_almost_equal (splist [- i ], arr [- i ])
96
112
97
113
98
114
if __name__ == '__main__' :
0 commit comments