def test_plot(geometry_nat, geometry_urb):
hd.interactive_map(geometry_nat.bounds)
nwis = NWIS()
qobs = nwis.get_streamflow([SID_NATURAL, SID_URBAN], DATES_LONG)
hd.plot.signatures(qobs, precipitation=qobs[f"USGS-{SID_NATURAL}"], output="data/hd.plot.png")
hd.plot.signatures(qobs[f"USGS-{SID_NATURAL}"], precipitation=qobs[f"USGS-{SID_NATURAL}"])
_, _, levels = hd.plot.cover_legends()
assert levels[-1] == 100
def test_nwis(geometry_nat):
nwis = NWIS()
qobs = nwis.get_streamflow(SID_NATURAL, DATES, mmd=True)
info = nwis.get_info(nwis.query_byid(SID_NATURAL), expanded=True)
info_box = nwis.get_info(nwis.query_bybox(geometry_nat.bounds))
assert (abs(qobs.sum().item() - 27.630) < 1e-3 and info.hcdn_2009.item()
and info_box.site_no.tolist()
== ["01031300", "01031450", "01031500"])