From 3666f3d418f1d8098007af584908eef4f5ce1b58 Mon Sep 17 00:00:00 2001 From: l-kotzur Date: Wed, 28 Jun 2023 15:34:59 +0200 Subject: [PATCH] add test for weeks instead of days as period --- test/test_hierarchical.py | 22 ++++++++++++++++++++++ 1 file changed, 22 insertions(+) diff --git a/test/test_hierarchical.py b/test/test_hierarchical.py index 40d2f88..ac74406 100644 --- a/test/test_hierarchical.py +++ b/test/test_hierarchical.py @@ -51,6 +51,28 @@ def test_hierarchical(): np.testing.assert_array_almost_equal(orig.values, test.values, decimal=4) +def test_hierarchical_for_weeks(): + + raw = pd.read_csv( + os.path.join(os.path.dirname(__file__), "..", "examples", "testdata.csv"), + index_col=0, + ) + + starttime = time.time() + + aggregation = tsam.TimeSeriesAggregation( + raw, + noTypicalPeriods=8, + hoursPerPeriod=24*7, + clusterMethod="hierarchical", + extremePeriodMethod="new_cluster_center", + addPeakMin=["T"], + addPeakMax=["Load"], + ) + + typPeriods = aggregation.createTypicalPeriods() + + print("Clustering took " + str(time.time() - starttime)) if __name__ == "__main__": test_hierarchical()