File tree 1 file changed +6
-4
lines changed
1 file changed +6
-4
lines changed Original file line number Diff line number Diff line change @@ -339,13 +339,15 @@ def test_async_epidata(self):
339
339
self .params_from_row (rows [0 ], source = 'covidcast' ),
340
340
self .params_from_row (rows [1 ], source = 'covidcast' )
341
341
]* 12 , batch_size = 10 )
342
- responses = [i [0 ] for i in test_output ]
343
- # check response is same as standard covidcast call, using 24 calls to test batch sizing
342
+ responses = [i [0 ]["epidata" ] for i in test_output ]
343
+ # check response is same as standard covidcast call (minus fields omitted by the api.php endpoint),
344
+ # using 24 calls to test batch sizing
345
+ ignore_fields = ["source" , "time_type" , "geo_type" ]
344
346
self .assertEqual (
345
347
responses ,
346
348
[
347
- Epidata .covidcast (** self .params_from_row (rows [0 ])),
348
- Epidata .covidcast (** self .params_from_row (rows [1 ])),
349
+ [{ k : v for k , v in row . items () if k not in ignore_fields } for row in Epidata .covidcast (** self .params_from_row (rows [0 ]))[ "epidata" ]] ,
350
+ [{ k : v for k , v in row . items () if k not in ignore_fields } for row in Epidata .covidcast (** self .params_from_row (rows [1 ]))[ "epidata" ]] ,
349
351
]* 12
350
352
)
351
353
You can’t perform that action at this time.
0 commit comments