Skip to content

Commit 65d1587

Browse files
committed
More test tweaks
1 parent cd665bd commit 65d1587

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

integrations/client/test_delphi_epidata.py

+3-3
Original file line numberDiff line numberDiff line change
@@ -342,12 +342,12 @@ def test_async_epidata(self):
342342
responses = [i[0]["epidata"] for i in test_output]
343343
# check response is same as standard covidcast call (minus fields omitted by the api.php endpoint),
344344
# using 24 calls to test batch sizing
345-
ignore_fields = ["source", "time_type", "geo_type"]
345+
ignore_fields = CovidcastTestRow._api_row_compatibility_ignore_fields
346346
self.assertEqual(
347347
responses,
348348
[
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"]],
351351
]*12
352352
)
353353

0 commit comments

Comments
 (0)