/
test_embedding.py
78 lines (64 loc) · 3.46 KB
/
test_embedding.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
"""Testing for time series embedding."""
# License : Apache 2.0
import numpy as np
import pytest
from numpy.testing import assert_almost_equal
from sklearn.exceptions import NotFittedError
from giotto.time_series import TakensEmbedding
from giotto.time_series import SlidingWindow
signal = np.asarray([np.sin(x / 2) + 2 for x in range(0, 20)])
signal_embedded_search = np.array([[2., 2.47942554],
[2.47942554, 2.84147098],
[2.84147098, 2.99749499],
[2.99749499, 2.90929743],
[2.90929743, 2.59847214],
[2.59847214, 2.14112001],
[2.14112001, 1.64921677],
[1.64921677, 1.2431975],
[1.2431975, 1.02246988],
[1.02246988, 1.04107573],
[1.04107573, 1.29445967],
[1.29445967, 1.7205845],
[1.7205845, 2.21511999],
[2.21511999, 2.6569866],
[2.6569866, 2.93799998],
[2.93799998, 2.98935825],
[2.98935825, 2.79848711],
[2.79848711, 2.41211849],
[2.41211849, 1.92484888]])
signal_embedded_fixed = \
np.array([[2., 2.47942554, 2.84147098, 2.99749499, 2.90929743],
[2.47942554, 2.84147098, 2.99749499, 2.90929743, 2.59847214],
[2.84147098, 2.99749499, 2.90929743, 2.59847214, 2.14112001],
[2.99749499, 2.90929743, 2.59847214, 2.14112001, 1.64921677],
[2.90929743, 2.59847214, 2.14112001, 1.64921677, 1.2431975],
[2.59847214, 2.14112001, 1.64921677, 1.2431975, 1.02246988],
[2.14112001, 1.64921677, 1.2431975, 1.02246988, 1.04107573],
[1.64921677, 1.2431975, 1.02246988, 1.04107573, 1.29445967],
[1.2431975, 1.02246988, 1.04107573, 1.29445967, 1.7205845],
[1.02246988, 1.04107573, 1.29445967, 1.7205845, 2.21511999],
[1.04107573, 1.29445967, 1.7205845, 2.21511999, 2.6569866],
[1.29445967, 1.7205845, 2.21511999, 2.6569866, 2.93799998],
[1.7205845, 2.21511999, 2.6569866, 2.93799998, 2.98935825],
[2.21511999, 2.6569866, 2.93799998, 2.98935825, 2.79848711],
[2.6569866, 2.93799998, 2.98935825, 2.79848711, 2.41211849],
[2.93799998, 2.98935825, 2.79848711, 2.41211849, 1.92484888]])
def test_embedder_params():
parameters_type = 'not_defined'
embedder = TakensEmbedding(parameters_type=parameters_type)
with pytest.raises(ValueError):
embedder.fit(signal)
def test_embedder_not_fitted():
embedder = TakensEmbedding()
with pytest.raises(NotFittedError):
embedder.transform(signal)
@pytest.mark.parametrize("parameters_type, expected",
[('search', signal_embedded_search),
('fixed', signal_embedded_fixed)])
def test_embedder_transform(parameters_type, expected):
embedder = TakensEmbedding(parameters_type=parameters_type)
assert_almost_equal(embedder.fit_transform(signal), expected)
def test_window_params():
window = SlidingWindow(width=0)
with pytest.raises(ValueError):
window.fit(signal)