File tree 1 file changed +5
-4
lines changed
1 file changed +5
-4
lines changed Original file line number Diff line number Diff line change @@ -335,9 +335,9 @@ 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
342
responses = [i [0 ] for i in test_output ]
343
343
# check response is same as standard covidcast call, using 24 calls to test batch sizing
@@ -352,8 +352,9 @@ def test_async_epidata(self):
352
352
@fake_epidata_endpoint
353
353
def test_async_epidata_fail (self ):
354
354
with pytest .raises (ClientResponseError , match = "404, message='NOT FOUND'" ):
355
- Epidata .async_epidata ('covidcast' , [
355
+ Epidata .async_epidata ([
356
356
{
357
+ 'source' : 'covidcast' ,
357
358
'data_source' : 'src' ,
358
359
'signals' : 'sig' ,
359
360
'time_type' : 'day' ,
You can’t perform that action at this time.
0 commit comments