Skip to content

Commit b223b4b

Browse files
committed
fix: /coverage tests
1 parent 42e040b commit b223b4b

File tree

2 files changed

+8
-5
lines changed

2 files changed

+8
-5
lines changed

integrations/server/test_covidcast_endpoints.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -369,18 +369,20 @@ def test_coverage(self):
369369
dates = [20200401 + i for i in range(len(num_geos_per_date))]
370370
rows = [CovidcastRow(time_value=dates[i], value=i, geo_value=str(geo_value)) for i, num_geo in enumerate(num_geos_per_date) for geo_value in range(num_geo)]
371371
self._insert_rows(rows)
372+
first = rows[0]
372373

373374
with self.subTest("default"):
374-
out = self._fetch("/coverage")
375+
out = self._fetch("/coverage", signal=first.signal_pair, latest=dates[-1], format="json")
375376
self.assertEqual(len(out), len(num_geos_per_date))
376377
self.assertEqual([o["time_value"] for o in out], dates)
377378
self.assertEqual([o["count"] for o in out], num_geos_per_date)
378379

379380
with self.subTest("specify window"):
380-
out = self._fetch("/coverage", window="20200401-20200402")
381+
out = self._fetch("/coverage", signal=first.signal_pair, window=f"{dates[0]}-{dates[1]}", format="json")
381382
self.assertEqual(len(out), 2)
382383
self.assertEqual([o["time_value"] for o in out], dates[:2])
383384
self.assertEqual([o["count"] for o in out], num_geos_per_date[:2])
385+
384386
with self.subTest("invalid geo_type"):
385-
out = self._fetch("/coverage", geo_type="state")
387+
out = self._fetch("/coverage", signal=first.signal_pair, geo_type="state", format="json")
386388
self.assertEqual(len(out), 0)

src/server/endpoints/covidcast.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,7 @@
3434
from .._db import sql_table_has_columns
3535
from .._pandas import as_pandas
3636
from .covidcast_utils import compute_trend, compute_trends, compute_correlations, compute_trend_value, CovidcastMetaEntry, AllSignalsMap
37-
from ..utils import shift_time_value, date_to_time_value, time_value_to_iso
37+
from ..utils import shift_time_value, date_to_time_value, time_value_to_iso, time_value_to_date
3838

3939
# first argument is the endpoint name
4040
bp = Blueprint("covidcast", __name__)
@@ -505,7 +505,8 @@ def handle_coverage():
505505
if "window" in request.values:
506506
time_window = parse_day_range_arg("window")
507507
else:
508-
now = date.today()
508+
now_time = extract_date("latest")
509+
now = date.today() if now_time is None else time_value_to_date(now_time)
509510
last = extract_integer("days")
510511
if last is None:
511512
last = 30

0 commit comments

Comments
 (0)