@@ -335,25 +335,28 @@ def test_async_epidata(self):
335
335
]
336
336
self ._insert_rows (rows )
337
337
338
- test_output = Epidata .async_epidata ('covidcast' , [
339
- self .params_from_row (rows [0 ]),
340
- self .params_from_row (rows [1 ])
338
+ test_output = Epidata .async_epidata ([
339
+ self .params_from_row (rows [0 ], source = 'covidcast' ),
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 = CovidcastTestRow ._api_row_compatibility_ignore_fields
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 : row [ k ] for k in row . keys () - ignore_fields } for row in Epidata .covidcast (** self .params_from_row (rows [0 ]))[ "epidata" ]] ,
350
+ [{ k : row [ k ] for k in row . keys () - ignore_fields } for row in Epidata .covidcast (** self .params_from_row (rows [1 ]))[ "epidata" ]] ,
349
351
]* 12
350
352
)
351
353
352
354
@fake_epidata_endpoint
353
355
def test_async_epidata_fail (self ):
354
356
with pytest .raises (ClientResponseError , match = "404, message='NOT FOUND'" ):
355
- Epidata .async_epidata ('covidcast' , [
357
+ Epidata .async_epidata ([
356
358
{
359
+ 'source' : 'covidcast' ,
357
360
'data_source' : 'src' ,
358
361
'signals' : 'sig' ,
359
362
'time_type' : 'day' ,
0 commit comments