File tree 1 file changed +3
-3
lines changed
1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -342,12 +342,12 @@ def test_async_epidata(self):
342
342
responses = [i [0 ]["epidata" ] for i in test_output ]
343
343
# check response is same as standard covidcast call (minus fields omitted by the api.php endpoint),
344
344
# using 24 calls to test batch sizing
345
- ignore_fields = [ "source" , "time_type" , "geo_type" ]
345
+ ignore_fields = CovidcastTestRow . _api_row_compatibility_ignore_fields
346
346
self .assertEqual (
347
347
responses ,
348
348
[
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
+ [{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" ]],
351
351
]* 12
352
352
)
353
353
You can’t perform that action at this time.
0 commit comments