From 057ee4c0635eabe3a5e699e50b665bc73a362377 Mon Sep 17 00:00:00 2001 From: Florian Pinault Date: Mon, 26 Feb 2024 10:51:20 +0000 Subject: [PATCH] clean test --- tests/sources/test_constants.py | 20 ++++++++++++-------- 1 file changed, 12 insertions(+), 8 deletions(-) diff --git a/tests/sources/test_constants.py b/tests/sources/test_constants.py index f57b11f6..a617a811 100644 --- a/tests/sources/test_constants.py +++ b/tests/sources/test_constants.py @@ -93,19 +93,23 @@ def test_constant_3(): date = sample[0].datetime() date_plus_6h = date + datetime.timedelta(hours=6) - a = load_source("constants", sample, date=date_plus_6h, param="insolation") - b = load_source("constants", sample, date=date, param="insolation+6h") + a = load_source( + "constants", sample, date=date_plus_6h, param="cos_solar_zenith_angle" + ) + b = load_source("constants", sample, date=date, param="cos_solar_zenith_angle+6h") assert np.all(a.to_numpy() == b.to_numpy()) - assert a.unique_values("param")["param"] == ("insolation",) - assert b.unique_values("param")["param"] == ("insolation+6h",) + assert a[0].metadata("param") == "cos_solar_zenith_angle" + assert b[0].metadata("param") == "cos_solar_zenith_angle+6h" date = sample[0].datetime() date_minus_30d = date + datetime.timedelta(days=-30) - a = load_source("constants", sample, date=date_minus_30d, param="insolation") - b = load_source("constants", sample, date=date, param="insolation-30d") + a = load_source( + "constants", sample, date=date_minus_30d, param="cos_solar_zenith_angle" + ) + b = load_source("constants", sample, date=date, param="cos_solar_zenith_angle-30d") assert np.all(a.to_numpy() == b.to_numpy()) - assert a.unique_values("param")["param"] == ("insolation",) - assert b.unique_values("param")["param"] == ("insolation-30d",) + assert a[0].metadata("param") == "cos_solar_zenith_angle" + assert b[0].metadata("param") == "cos_solar_zenith_angle-30d" if __name__ == "__main__":