@@ -17,7 +17,7 @@ def setUp(self):
1717 for day in range (4 )
1818 for hour in range (24 )
1919 ],
20- 'value' : list ( range ( 4 * 24 ) )
20+ 'value' : np . arange ( 0 , 4 * 24 , dtype = float )
2121 })
2222
2323 def test_resample_rule_str (self ):
@@ -50,22 +50,22 @@ def test_resample_groupby(self):
5050 out = resample (self .df .set_index ('dt' ), '1d' , groupby = ['group1' , 'group2' ])
5151
5252 assert_frame_equal (out , pd .DataFrame ([
53- {'group1' : 'A' , 'group2' : 'C' , 'dt' : datetime (2000 , 1 , 1 ), 'value' : 10 },
54- {'group1' : 'A' , 'group2' : 'C' , 'dt' : datetime (2000 , 1 , 2 ), 'value' : 34 },
55- {'group1' : 'A' , 'group2' : 'C' , 'dt' : datetime (2000 , 1 , 3 ), 'value' : 58 },
56- {'group1' : 'A' , 'group2' : 'C' , 'dt' : datetime (2000 , 1 , 4 ), 'value' : 82 },
57- {'group1' : 'A' , 'group2' : 'D' , 'dt' : datetime (2000 , 1 , 1 ), 'value' : 12 },
58- {'group1' : 'A' , 'group2' : 'D' , 'dt' : datetime (2000 , 1 , 2 ), 'value' : 36 },
59- {'group1' : 'A' , 'group2' : 'D' , 'dt' : datetime (2000 , 1 , 3 ), 'value' : 60 },
60- {'group1' : 'A' , 'group2' : 'D' , 'dt' : datetime (2000 , 1 , 4 ), 'value' : 84 },
61- {'group1' : 'B' , 'group2' : 'C' , 'dt' : datetime (2000 , 1 , 1 ), 'value' : 11 },
62- {'group1' : 'B' , 'group2' : 'C' , 'dt' : datetime (2000 , 1 , 2 ), 'value' : 35 },
63- {'group1' : 'B' , 'group2' : 'C' , 'dt' : datetime (2000 , 1 , 3 ), 'value' : 59 },
64- {'group1' : 'B' , 'group2' : 'C' , 'dt' : datetime (2000 , 1 , 4 ), 'value' : 83 },
65- {'group1' : 'B' , 'group2' : 'D' , 'dt' : datetime (2000 , 1 , 1 ), 'value' : 13 },
66- {'group1' : 'B' , 'group2' : 'D' , 'dt' : datetime (2000 , 1 , 2 ), 'value' : 37 },
67- {'group1' : 'B' , 'group2' : 'D' , 'dt' : datetime (2000 , 1 , 3 ), 'value' : 61 },
68- {'group1' : 'B' , 'group2' : 'D' , 'dt' : datetime (2000 , 1 , 4 ), 'value' : 85 },
53+ {'group1' : 'A' , 'group2' : 'C' , 'dt' : datetime (2000 , 1 , 1 ), 'value' : 10.0 },
54+ {'group1' : 'A' , 'group2' : 'C' , 'dt' : datetime (2000 , 1 , 2 ), 'value' : 34.0 },
55+ {'group1' : 'A' , 'group2' : 'C' , 'dt' : datetime (2000 , 1 , 3 ), 'value' : 58.0 },
56+ {'group1' : 'A' , 'group2' : 'C' , 'dt' : datetime (2000 , 1 , 4 ), 'value' : 82.0 },
57+ {'group1' : 'A' , 'group2' : 'D' , 'dt' : datetime (2000 , 1 , 1 ), 'value' : 12.0 },
58+ {'group1' : 'A' , 'group2' : 'D' , 'dt' : datetime (2000 , 1 , 2 ), 'value' : 36.0 },
59+ {'group1' : 'A' , 'group2' : 'D' , 'dt' : datetime (2000 , 1 , 3 ), 'value' : 60.0 },
60+ {'group1' : 'A' , 'group2' : 'D' , 'dt' : datetime (2000 , 1 , 4 ), 'value' : 84.0 },
61+ {'group1' : 'B' , 'group2' : 'C' , 'dt' : datetime (2000 , 1 , 1 ), 'value' : 11.0 },
62+ {'group1' : 'B' , 'group2' : 'C' , 'dt' : datetime (2000 , 1 , 2 ), 'value' : 35.0 },
63+ {'group1' : 'B' , 'group2' : 'C' , 'dt' : datetime (2000 , 1 , 3 ), 'value' : 59.0 },
64+ {'group1' : 'B' , 'group2' : 'C' , 'dt' : datetime (2000 , 1 , 4 ), 'value' : 83.0 },
65+ {'group1' : 'B' , 'group2' : 'D' , 'dt' : datetime (2000 , 1 , 1 ), 'value' : 13.0 },
66+ {'group1' : 'B' , 'group2' : 'D' , 'dt' : datetime (2000 , 1 , 2 ), 'value' : 37.0 },
67+ {'group1' : 'B' , 'group2' : 'D' , 'dt' : datetime (2000 , 1 , 3 ), 'value' : 61.0 },
68+ {'group1' : 'B' , 'group2' : 'D' , 'dt' : datetime (2000 , 1 , 4 ), 'value' : 85.0 },
6969 ], columns = ['group1' , 'group2' , 'dt' , 'value' ]))
7070
7171 def test_resample_on (self ):
@@ -95,30 +95,30 @@ def test_resample_aggregation_str(self):
9595 out = resample (self .df .set_index ('dt' ), '1d' , aggregation = 'max' )
9696
9797 assert_frame_equal (out , pd .DataFrame ([
98- {'dt' : datetime (2000 , 1 , 1 ), 'value' : 23 },
99- {'dt' : datetime (2000 , 1 , 2 ), 'value' : 47 },
100- {'dt' : datetime (2000 , 1 , 3 ), 'value' : 71 },
101- {'dt' : datetime (2000 , 1 , 4 ), 'value' : 95 },
98+ {'dt' : datetime (2000 , 1 , 1 ), 'value' : 23.0 },
99+ {'dt' : datetime (2000 , 1 , 2 ), 'value' : 47.0 },
100+ {'dt' : datetime (2000 , 1 , 3 ), 'value' : 71.0 },
101+ {'dt' : datetime (2000 , 1 , 4 ), 'value' : 95.0 },
102102 ]))
103103
104104 def test_resample_aggregation_func (self ):
105105
106106 out = resample (self .df .set_index ('dt' ), '1d' , aggregation = np .max )
107107
108108 assert_frame_equal (out , pd .DataFrame ([
109- {'dt' : datetime (2000 , 1 , 1 ), 'value' : 23 },
110- {'dt' : datetime (2000 , 1 , 2 ), 'value' : 47 },
111- {'dt' : datetime (2000 , 1 , 3 ), 'value' : 71 },
112- {'dt' : datetime (2000 , 1 , 4 ), 'value' : 95 },
109+ {'dt' : datetime (2000 , 1 , 1 ), 'value' : 23.0 },
110+ {'dt' : datetime (2000 , 1 , 2 ), 'value' : 47.0 },
111+ {'dt' : datetime (2000 , 1 , 3 ), 'value' : 71.0 },
112+ {'dt' : datetime (2000 , 1 , 4 ), 'value' : 95.0 },
113113 ]))
114114
115115 def test_resample_aggregation_import (self ):
116116
117117 out = resample (self .df .set_index ('dt' ), '1d' , aggregation = 'numpy.max' )
118118
119119 assert_frame_equal (out , pd .DataFrame ([
120- {'dt' : datetime (2000 , 1 , 1 ), 'value' : 23 },
121- {'dt' : datetime (2000 , 1 , 2 ), 'value' : 47 },
122- {'dt' : datetime (2000 , 1 , 3 ), 'value' : 71 },
123- {'dt' : datetime (2000 , 1 , 4 ), 'value' : 95 },
120+ {'dt' : datetime (2000 , 1 , 1 ), 'value' : 23.0 },
121+ {'dt' : datetime (2000 , 1 , 2 ), 'value' : 47.0 },
122+ {'dt' : datetime (2000 , 1 , 3 ), 'value' : 71.0 },
123+ {'dt' : datetime (2000 , 1 , 4 ), 'value' : 95.0 },
124124 ]))
0 commit comments