Skip to content

Commit 9d6394a

Browse files
committed
update tests to process to include wip and 7d_avg signals
1 parent 509f4d7 commit 9d6394a

File tree

2 files changed

+48
-20
lines changed

2 files changed

+48
-20
lines changed
Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,2 +1,4 @@
11
origin_census_block_group,date_range_start,date_range_end,device_count,completely_home_device_count,median_home_dwell_time,part_time_work_behavior_devices,full_time_work_behavior_devices
2-
10539707003,2020-06-12T00:00:00-05:00,2020-06-13T00:00:00-05:00,100,15,6,35,45
2+
10539707003,2020-06-12T00:00:00-05:00,2020-06-13T00:00:00-05:00,100,15,6,35,45
3+
131510702031,2020-06-12T00:00:00-05:00,2020-06-13T00:00:00-05:00,100,5,3,5,5
4+
131510702031,2020-06-12T00:00:00-05:00,2020-06-13T00:00:00-05:00,100,6,4,6,6

safegraph/tests/test_process.py

Lines changed: 45 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -143,37 +143,63 @@ def test_process(self, tmp_path):
143143
'raw_data/small_raw_data_2.csv',
144144
'raw_data/small_raw_data_3.csv', ],
145145
SIGNALS,
146+
['median_home_dwell_time',
147+
'completely_home_prop_7d_avg'],
146148
['state'],
147149
export_dir)
148150

149151
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]
155157
}),
156158
'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]
161163
}),
162164
'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]
167169
}),
168170
'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]
173199
})
174200
}
175201
actual = {signal: pd.read_csv(
176202
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:
179205
pd.testing.assert_frame_equal(expected[signal], actual[signal])

0 commit comments

Comments
 (0)