start unit tests on new functions

Author: ehinman

tests/waterdata_test.py

@@ -6,6 +6,11 @@
 from dataretrieval.waterdata import (
     _check_profiles,
     get_samples,
+    get_daily,
+    get_monitoring_locations,
+    get_latest_continuous,
+    get_field_measurements,
+    get_time_series_metadata,
     _SERVICES,
     _PROFILES
 )
@@ -105,3 +110,88 @@ def test_samples_organizations():
         )
     assert len(df) == 1
     assert df.size == 3
+
+def test_get_daily():
+    df = get_daily(
+        monitoring_location_id="USGS-05427718",
+        parameter_code="00060",
+        time="2025-01-01/.."
+    )
+    assert "daily_id" in df.columns
+    assert "geometry" in df.columns
+    assert df.shape[1] == 12
+    assert df.parameter_code.unique().tolist() == ["00060"]
+    assert df.monitoring_location_id.unique().tolist() == ["USGS-05427718"]
+    assert df["time"].apply(lambda x: isinstance(x, datetime.date)).all()
+    assert df["value"].dtype == "float64"
+
+def test_get_daily_properties():
+    df = get_daily(
+        monitoring_location_id="USGS-05427718",
+        parameter_code="00060",
+        time="2025-01-01/..",
+        properties=["daily_id", "monitoring_location_id", "parameter_code", "time", "value", "geometry"]
+    )
+    assert "daily_id" in df.columns
+    assert "geometry" in df.columns
+    assert df.shape[1] == 6
+    assert (df["time"] >= datetime.date(2025, 1, 1)).all()
+
+def test_get_daily_no_geometry():
+    df = get_daily(
+        monitoring_location_id="USGS-05427718",
+        parameter_code="00060",
+        time="2025-01-01/..",
+        skipGeometry=True
+    )
+    assert "geometry" not in df.columns
+    assert df.shape[1] == 11
+    assert isinstance(df, DataFrame)
+
+def test_get_monitoring_locations():
+    df = get_monitoring_locations(
+        state_name="Connecticut",
+        site_type_code="GW"
+    )
+    assert df.site_type_code.unique().tolist() == ["GW"]
+
+def test_get_monitoring_locations_hucs():
+    df = get_monitoring_locations(
+        hydrologic_unit_code=["010802050102", "010802050103"]
+    )
+    assert set(df.hydrologic_unit_code.unique().tolist()) == {"010802050102", "010802050103"}
+
+def test_get_latest_continuous():
+    df = get_latest_continuous(
+        monitoring_location_id=["USGS-05427718", "USGS-05427719"],
+        parameter_code=["00060", "00065"]
+    )
+    assert df.shape[0] <= 4
+    assert df.statistic_id.unique().tolist() == ["00011"]
+    try:
+        datetime.datetime.strptime(df['time'].iloc[0], "%Y-%m-%dT%H:%M:%S+00:00")
+        out=True
+    except:
+        out=False
+    assert out
+
+def test_get_field_measurements():
+    df = get_field_measurements(
+        monitoring_location_id="USGS-05427718",
+        unit_of_measure="ft^3/s",
+        time="2025-01-01/2025-10-01",
+        skipGeometry=True
+    )
+    assert "field_measurement_id" in df.columns
+    assert "geometry" not in df.columns
+    assert df.unit_of_measure.unique().tolist() == ["ft^3/s"]
+
+def test_get_time_series_metadata():
+    df = get_time_series_metadata(
+        bbox=[-89.840355,42.853411,-88.818626,43.422598],
+        parameter_code=["00060", "00065", "72019"],
+        skipGeometry=True
+    )
+    assert set(df['parameter_name'].unique().tolist()) == {"Gage height", "Water level, depth LSD", "Discharge"}
+
+