@@ -369,18 +369,20 @@ def test_coverage(self):
369
369
dates = [20200401 + i for i in range (len (num_geos_per_date ))]
370
370
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 )]
371
371
self ._insert_rows (rows )
372
+ first = rows [0 ]
372
373
373
374
with self .subTest ("default" ):
374
- out = self ._fetch ("/coverage" )
375
+ out = self ._fetch ("/coverage" , signal = first . signal_pair , latest = dates [ - 1 ], format = "json" )
375
376
self .assertEqual (len (out ), len (num_geos_per_date ))
376
377
self .assertEqual ([o ["time_value" ] for o in out ], dates )
377
378
self .assertEqual ([o ["count" ] for o in out ], num_geos_per_date )
378
379
379
380
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 " )
381
382
self .assertEqual (len (out ), 2 )
382
383
self .assertEqual ([o ["time_value" ] for o in out ], dates [:2 ])
383
384
self .assertEqual ([o ["count" ] for o in out ], num_geos_per_date [:2 ])
385
+
384
386
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 " )
386
388
self .assertEqual (len (out ), 0 )
0 commit comments