@@ -143,37 +143,63 @@ def test_process(self, tmp_path):
143
143
'raw_data/small_raw_data_2.csv' ,
144
144
'raw_data/small_raw_data_3.csv' , ],
145
145
SIGNALS ,
146
+ ['median_home_dwell_time' ,
147
+ 'completely_home_prop_7d_avg' ],
146
148
['state' ],
147
149
export_dir )
148
150
149
151
expected = {
150
- 'median_home_dwell_time ' : pd .DataFrame (data = {
151
- 'geo_id' : ['al' , 'pa ' ],
152
- 'val' : [4.5 , 7 .5 ],
153
- 'se' : [1.5 , 0.5 ],
154
- 'sample_size' : [2 , 2 ]
152
+ 'wip_median_home_dwell_time ' : pd .DataFrame (data = {
153
+ 'geo_id' : ['al' , 'ga ' ],
154
+ 'val' : [6 , 3 .5 ],
155
+ 'se' : [None , 0.5 ],
156
+ 'sample_size' : [1 , 2 ]
155
157
}),
156
158
'completely_home_prop' : pd .DataFrame (data = {
157
- 'geo_id' : ['al' , 'pa ' ],
158
- 'val' : [0.1 , 0.15 ],
159
- 'se' : [0.05 , 0.05 ],
160
- 'sample_size' : [2 , 2 ]
159
+ 'geo_id' : ['al' , 'ga ' ],
160
+ 'val' : [0.15 , 0.055 ],
161
+ 'se' : [None , 0.005 ],
162
+ 'sample_size' : [1 , 2 ]
161
163
}),
162
164
'part_time_work_prop' : pd .DataFrame (data = {
163
- 'geo_id' : ['al' , 'pa ' ],
164
- 'val' : [0.25 , 0.25 ],
165
- 'se' : [0.1 , 0.05 ],
166
- 'sample_size' : [2 , 2 ]
165
+ 'geo_id' : ['al' , 'ga ' ],
166
+ 'val' : [0.35 , 0.055 ],
167
+ 'se' : [None , 0.005 ],
168
+ 'sample_size' : [1 , 2 ]
167
169
}),
168
170
'full_time_work_prop' : pd .DataFrame (data = {
169
- 'geo_id' : ['al' , 'pa' ],
170
- 'val' : [0.35 , 0.35 ],
171
- 'se' : [0.1 , 0.05 ],
172
- 'sample_size' : [2 , 2 ]
171
+ 'geo_id' : ['al' , 'ga' ],
172
+ 'val' : [0.45 , 0.055 ],
173
+ 'se' : [None , 0.005 ],
174
+ 'sample_size' : [1 , 2 ]
175
+ }),
176
+ 'median_home_dwell_time_7d_avg' : pd .DataFrame (data = {
177
+ 'geo_id' : ['al' , 'ga' , 'pa' ],
178
+ 'val' : [4.5 , 3.5 , 7.5 ],
179
+ 'se' : [1.5 , 0.5 , 0.5 ],
180
+ 'sample_size' : [2 , 2 , 2 ]
181
+ }),
182
+ 'wip_completely_home_prop_7d_avg' : pd .DataFrame (data = {
183
+ 'geo_id' : ['al' , 'ga' , 'pa' ],
184
+ 'val' : [0.1 , 0.055 , 0.15 ],
185
+ 'se' : [0.05 , 0.005 , 0.05 ],
186
+ 'sample_size' : [2 , 2 , 2 ]
187
+ }),
188
+ 'part_time_work_prop_7d_avg' : pd .DataFrame (data = {
189
+ 'geo_id' : ['al' , 'ga' , 'pa' ],
190
+ 'val' : [0.25 , 0.055 , 0.25 ],
191
+ 'se' : [0.1 , 0.005 , 0.05 ],
192
+ 'sample_size' : [2 , 2 , 2 ]
193
+ }),
194
+ 'full_time_work_prop_7d_avg' : pd .DataFrame (data = {
195
+ 'geo_id' : ['al' , 'ga' , 'pa' ],
196
+ 'val' : [0.35 , 0.055 , 0.35 ],
197
+ 'se' : [0.1 , 0.005 , 0.05 ],
198
+ 'sample_size' : [2 , 2 , 2 ]
173
199
})
174
200
}
175
201
actual = {signal : pd .read_csv (
176
202
export_dir / f'2020-06-12_state_{ signal } .csv' )
177
- for signal in SIGNALS }
178
- for signal in SIGNALS :
203
+ for signal in expected }
204
+ for signal in expected :
179
205
pd .testing .assert_frame_equal (expected [signal ], actual [signal ])
0 commit comments