def test_pytwed_slow_2D(self):
s1 = np.array([[9.868, 14.54], [5.235, 16.3], [6.203, 16.76],
[7.968, 17.13], [6.493, 17.58], [12.466, 16.68],
[13.255, 14.31], [11.263, 12.09], [7.155, 10.53],
[7.377, 9.48], [8.672, 8.73], [7.881,
8.58]]).astype('float64')
s2 = np.array([[9.428, 5.06], [12.91, 5.78], [9.365, 7.01],
[12.535, 7.99], [10.152, 8.89], [11.473, 9.58],
[14.701, 8.12], [10.156, 6.9], [8.292, 5.82],
[5.946, 4.96], [6.339, 4.09], [5.395,
4.]]).astype('float64')
s3 = np.array([[7.371, 11.75], [5.35, 12.91], [5.101, 14.03],
[5.088, 14.44], [8.003, 14.52], [12.372, 13.86],
[14.212, 13.17], [9.517, 11.77], [8.634, 11.17],
[10.243, 11.2], [6.843, 10.9],
[4.888, 10.38]]).astype('float64')
ts1 = np.arange(len(s1))
ts2 = np.arange(len(s2))
ts3 = np.arange(len(s3))
nu = .1
lmbda = .2
p = 2
assert_almost_equal(twed(s1,
s2,
ts1,
ts2,
nu=nu,
lmbda=lmbda,
p=p,
fast=False),
74.222,
decimal=3)
assert_almost_equal(twed(s1, s2, nu=nu, lmbda=lmbda, p=p, fast=False),
74.222,
decimal=3)
assert_almost_equal(twed(s1,
s3,
ts1,
ts3,
nu=nu,
lmbda=lmbda,
p=p,
fast=False),
47.325,
decimal=3)
assert_almost_equal(twed(s2,
s3,
ts2,
ts3,
nu=nu,
lmbda=lmbda,
p=p,
fast=False),
69.047,
decimal=3)
def test_pytwed_flat_slow(self):
nu = .001
lmbda = 1.0
p = 2
s4 = np.array([1.0, 2.1, 3.2, 4.3, 5.4]).astype('float64')
s5 = np.array([9.8, 7.6, 6.5, 4.3, 2.1]).astype('float64')
ts4 = np.array([1., 2., 3., 4., 5.]).astype('float64')
ts5 = np.array([1., 2., 3., 4., 5.]).astype('float64')
assert_almost_equal(twed(s4,
s5,
ts4,
ts5,
nu=nu,
lmbda=lmbda,
p=p,
fast=False),
24.904,
decimal=3)
def test_pytwed_fast(self):
s1 = np.array([[0.], [0.], [1.], [1.], [2.], [3.], [5.], [2.], [0.],
[1.], [-0.1]]).astype('float64')
s2 = np.array([[0.], [1.], [2.], [2.5], [3.], [3.5], [4.], [4.5],
[5.5], [2.], [0.], [0.], [.25], [.05],
[0.]]).astype('float64')
s3 = np.array([[4.], [4.], [3.], [3.], [3.], [3.], [2.], [5.], [2.],
[.5], [.5], [.5]]).astype('float64')
ts1 = np.arange(len(s1))
ts2 = np.arange(len(s2))
ts3 = np.arange(len(s3))
nu = .1
lmbda = .2
p = 2
assert_almost_equal(twed(s1,
s2,
ts1,
ts2,
nu=nu,
lmbda=lmbda,
p=p,
fast=True),
11.9,
decimal=5) # was 6.250000000000001 in TWED
assert_almost_equal(twed(s1, s2, nu=nu, lmbda=lmbda, p=p, fast=True),
11.9,
decimal=5)
assert_almost_equal(twed(s1,
s3,
ts1,
ts3,
nu=nu,
lmbda=lmbda,
p=p,
fast=True),
16.3,
decimal=5) # was 25.159999999999993 in TWED
assert_almost_equal(twed(s2,
s3,
ts2,
ts3,
nu=nu,
lmbda=lmbda,
p=p,
fast=True),
19.9,
decimal=5) # was 28.529999999999994 in TWED
nu = .001
lmbda = 1.0
p = 2
s4 = np.array([[1.0], [2.1], [3.2], [4.3], [5.4]]).astype('float64')
s5 = np.array([[9.8], [7.6], [6.5], [4.3], [2.1]]).astype('float64')
ts4 = np.array([1., 2., 3., 4., 5.]).astype('float64')
ts5 = np.array([1., 2., 3., 4., 5.]).astype('float64')
assert_almost_equal(twed(s4,
s5,
ts4,
ts5,
nu=nu,
lmbda=lmbda,
p=p,
fast=True),
24.904,
decimal=3) # 24.903997...
def test_pytwed_slow(self):
s1 = np.array([[0.], [0.], [1.], [1.], [2.], [3.], [5.], [2.], [0.],
[1.], [-0.1]]).astype('float64')
s2 = np.array([[0.], [1.], [2.], [2.5], [3.], [3.5], [4.], [4.5],
[5.5], [2.], [0.], [0.], [.25], [.05],
[0.]]).astype('float64')
s3 = np.array([[4.], [4.], [3.], [3.], [3.], [3.], [2.], [5.], [2.],
[.5], [.5], [.5]]).astype('float64')
ts1 = np.arange(len(s1))
ts2 = np.arange(len(s2))
ts3 = np.arange(len(s3))
nu = .1
lmbda = .2
p = 2
assert_almost_equal(twed(s1,
s2,
ts1,
ts2,
nu=nu,
lmbda=lmbda,
p=p,
fast=False),
11.9,
decimal=5)
assert_almost_equal(twed(s1, s2, nu=nu, lmbda=lmbda, p=p, fast=False),
11.9,
decimal=5)
assert_almost_equal(twed(s1,
s3,
ts1,
ts3,
nu=nu,
lmbda=lmbda,
p=p,
fast=False),
16.3,
decimal=5)
assert_almost_equal(twed(s2,
s3,
ts2,
ts3,
nu=nu,
lmbda=lmbda,
p=p,
fast=False),
19.9,
decimal=5)
nu = .001
lmbda = 1.0
p = 2
s4 = np.array([[1.0], [2.1], [3.2], [4.3], [5.4]]).astype('float64')
s5 = np.array([[9.8], [7.6], [6.5], [4.3], [2.1]]).astype('float64')
ts4 = np.array([1., 2., 3., 4., 5.]).astype('float64')
ts5 = np.array([1., 2., 3., 4., 5.]).astype('float64')
assert_almost_equal(twed(s4,
s5,
ts4,
ts5,
nu=nu,
lmbda=lmbda,
p=p,
fast=False),
24.904,
decimal=3) # 24.904 exactly