def test_multivariate_timeseries_transformation_ok():
mv_timeseries = MultivariateTimeseries(
[[1, 2, 3, 4, 5], [0, 1, 2, 3, 4]], [6, 7, 8, 9, 10], ['foo', 'bar'], 'baz'
)
matrix, output = mv_timeseries.convert_to_snapshot_matrix(2)
assert_array_equal(
matrix,
array([[[1, 0], [2, 1], [3, 2]], [[2, 1], [3, 2], [4, 3]], [[3, 2], [4, 3], [5, 4]]]),
)
assert_array_equal(output, array([8, 9, 10]))
def test_mutlivariate_timeseries_transformation_from_dataframe_ok():
mv_timeseries_df = DataFrame({'foo': [1, 2, 3], 'bar': [4, 5, 6], 'baz': [0, 0, 0]})
mv_timeseries = MultivariateTimeseries.from_dataframe(mv_timeseries_df, ['foo', 'bar'], 'baz')
assert mv_timeseries
assert mv_timeseries.input_timeseries_collection[0].sequence == [1, 2, 3]
assert mv_timeseries.input_timeseries_collection[1].sequence == [4, 5, 6]
assert mv_timeseries.output_timeseries.sequence == [0, 0, 0]