diff --git a/test/time_feature/__init__.py b/test/time_feature/__init__.py new file mode 100644 index 0000000000..e69de29bb2 diff --git a/test/time_feature/test_base.py b/test/time_feature/test_base.py index a5c4e1dbc7..e448156b2b 100644 --- a/test/time_feature/test_base.py +++ b/test/time_feature/test_base.py @@ -16,7 +16,7 @@ from gluonts.time_feature import norm_freq_str -from common import M, Q, S, Y +from .common import M, Q, S, Y @pytest.mark.parametrize( diff --git a/test/time_feature/test_lag.py b/test/time_feature/test_lag.py index 650267564b..ad934bbd1d 100644 --- a/test/time_feature/test_lag.py +++ b/test/time_feature/test_lag.py @@ -19,7 +19,7 @@ import gluonts.time_feature.lag as date_feature_set -from common import H, M, Q, Y +from .common import H, M, Q, Y # These are the expected lags for common frequencies and corner cases. # By default all frequencies have the following lags: [1, 2, 3, 4, 5, 6, 7]. @@ -210,8 +210,7 @@ + [335, 336, 337, 503, 504, 505, 671, 672, 673, 719, 720, 721], # centered around each of the last 7 days (delta = 1) + last 4 weeks (delta = 1) + last 1 month (delta = 1) + # last 8th and 12th weeks (delta = 0) - "6" - + H: [ + ("6" + H): [ 1, 2, 3, @@ -242,24 +241,21 @@ + [224, 336], # centered around each of the last 7 days (delta = 1) + last 4 weeks (delta = 1) + last 1 month (delta = 1) + # last 8th and 12th weeks (delta = 0) + last year (delta = 1) - "12" - + H: [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15] + "12" + H: [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15] + [27, 28, 29, 41, 42, 43, 55, 56, 57] + [59, 60, 61] + [112, 168] + [727, 728, 729], # centered around each of the last 7 days (delta = 1) + last 4 weeks (delta = 1) + last 1 month (delta = 1) + # last 8th and 12th weeks (delta = 0) + last 3 years (delta = 1) - "23" - + H: [1, 2, 3, 4, 5, 6, 7, 8] + "23" + H: [1, 2, 3, 4, 5, 6, 7, 8] + [13, 14, 15, 20, 21, 22, 28, 29] + [30, 31, 32] + [58, 87] + [378, 379, 380, 758, 759, 760, 1138, 1139, 1140], # centered around each of the last 7 days (delta = 1) + last 4 weeks (delta = 1) + last 1 month (delta = 1) + # last 8th and 12th weeks (delta = 0) + last 3 years (delta = 1) - "25" - + H: [1, 2, 3, 4, 5, 6, 7] + "25" + H: [1, 2, 3, 4, 5, 6, 7] + [12, 13, 14, 19, 20, 21, 25, 26, 27] + [28, 29] + [53, 80] diff --git a/test/time_feature/test_seasonality.py b/test/time_feature/test_seasonality.py index a235dc1c35..3817416c2c 100644 --- a/test/time_feature/test_seasonality.py +++ b/test/time_feature/test_seasonality.py @@ -15,7 +15,7 @@ from gluonts.time_feature import get_seasonality -from common import H, M, Q, Y +from .common import H, M, Q, Y TEST_CASES = [ ("30min", 48),