def test_cwt():
ds = datasets._get_continuous()
# 1d
y = cwt_morlet(ds['x1'], [4, 6, 8])
assert y.ndim == 2
# 2d
y = cwt_morlet(ds['x2'], [4, 6, 8])
assert y.ndim == 3
def test_cwt():
ds = datasets._get_continuous()
# 1d
y = cwt_morlet(ds['x1'], [4, 6, 8])
assert y.ndim == 2
# 2d
y = cwt_morlet(ds['x2'], [4, 6, 8])
assert y.ndim == 3
def test_cwt():
"Test tests with wavelet transform"
ds = datasets.get_uts(True)
ds['cwt'] = cwt_morlet(ds['utsnd'], np.arange(10, 20))
res = testnd.ttest_rel('cwt', 'A', match='rm', ds=ds, pmin=0.05, samples=10)
cluster = res.clusters.sub("p == 0")
assert_array_equal(cluster['frequency_min'], 10)
assert_array_equal(cluster['frequency_max'], 19)
def test_cwt():
"Test tests with wavelet transform"
ds = datasets.get_uts(True)
ds['cwt'] = cwt_morlet(ds['utsnd'], np.arange(10, 20))
res = testnd.ttest_rel('cwt', 'A', match='rm', ds=ds, pmin=0.05, samples=10)
cluster = res.clusters.sub("p == 0")
assert_array_equal(cluster['frequency_min'], 10)
assert_array_equal(cluster['frequency_max'], 19)
def test_ndvar_indexing():
ds = datasets.get_uts(utsnd=True)
x = ds['utsnd']
# case
test_ndvar_index(x, 'case', 1, 1)
test_ndvar_index(x, 'case', [0, 3], [0, 3])
test_ndvar_index(x, 'case', slice(0, 10, 2), slice(0, 10, 2))
# sensor
test_ndvar_index(x, 'sensor', '0', 0)
test_ndvar_index(x, 'sensor', ['0', '2'], [0, 2])
test_ndvar_index(x, 'sensor', slice('0', '2'), slice(0, 2))
test_ndvar_index(x, 'sensor', 0, 0, False)
test_ndvar_index(x, 'sensor', [0, 2], [0, 2], False)
test_ndvar_index(x, 'sensor', slice(0, 2), slice(0, 2), False)
# time
test_ndvar_index(x, 'time', 0, 20)
test_ndvar_index(x, 'time', 0.1, 30)
test_ndvar_index(x, 'time', 0.102, 30, False)
test_ndvar_index(x, 'time', [0, 0.1, 0.2], [20, 30, 40])
test_ndvar_index(x, 'time', slice(0.1, None), slice(30, None))
test_ndvar_index(x, 'time', slice(0.2), slice(40))
test_ndvar_index(x, 'time', slice(0.202), slice(41), False)
test_ndvar_index(x, 'time', slice(0.1, 0.2), slice(30, 40))
test_ndvar_index(x, 'time', slice(0.102, 0.2), slice(31, 40), False)
test_ndvar_index(x, 'time', slice(0.1, None, 0.1), slice(30, None, 10))
test_ndvar_index(x, 'time', slice(0.1, None, 1), slice(30, None, 100))
# Ordered
x = cwt_morlet(ds['uts'], [8, 10, 13, 17])
assert_raises(IndexError, x.__getitem__, (full_slice, 9))
assert_raises(IndexError, x.__getitem__, (full_slice, 6))
test_ndvar_index(x, 'frequency', 10, 1)
test_ndvar_index(x, 'frequency', 10.1, 1, False)
test_ndvar_index(x, 'frequency', 9.9, 1, False)
test_ndvar_index(x, 'frequency', [8.1, 10.1], [0, 1], False)
test_ndvar_index(x, 'frequency', slice(8, 13), slice(0, 2))
test_ndvar_index(x, 'frequency', slice(8, 13.1), slice(0, 3), False)
test_ndvar_index(x, 'frequency', slice(8, 13.1, 2), slice(0, 3, 2), False)
# Categorial
x = NDVar(x.x, ('case', Categorial('cat', ['8', '10', '13', '17']), x.time))
assert_raises(TypeError, x.__getitem__, (full_slice, 9))
assert_raises(IndexError, x.__getitem__, (full_slice, '9'))
test_ndvar_index(x, 'cat', '13', 2)
test_ndvar_index(x, 'cat', ['8', '13'], [0, 2])
test_ndvar_index(x, 'cat', slice('8', '13'), slice(0, 2))
test_ndvar_index(x, 'cat', slice('8', None, 2), slice(0, None, 2))
# SourceSpace
x = datasets.get_mne_stc(True)
assert_raises(TypeError, x.__getitem__, slice('insula-rh'))
assert_raises(TypeError, x.__getitem__, slice('insula-lh', 'insula-rh'))
assert_raises(TypeError, x.__getitem__, ('insula-lh', 'insula-rh'))
test_ndvar_index(x, 'source', 'L90', 90)
test_ndvar_index(x, 'source', 'R90', 642 + 90)
test_ndvar_index(x, 'source', ['L90', 'R90'], [90, 642 + 90])
test_ndvar_index(x, 'source', slice('L90', 'R90'), slice(90, 642 + 90))
test_ndvar_index(x, 'source', 90, 90, False)
test_ndvar_index(x, 'source', [90, 95], [90, 95], False)
test_ndvar_index(x, 'source', slice(90, 95), slice(90, 95), False)
test_ndvar_index(x, 'source', 'insula-lh', x.source.parc == 'insula-lh', False)
test_ndvar_index(x, 'source', ('insula-lh', 'insula-rh'),
x.source.parc.isin(('insula-lh', 'insula-rh')), False)
n_lh = x.source.parc.endswith('lh').sum()
test_ndvar_index(x, 'source', 'lh', slice(n_lh), False)
test_ndvar_index(x, 'source', 'rh', slice(n_lh, None), False)
# argmax
x.x[10, 10] = 20
eq_(x.argmax(), ('L10', 0.1))
eq_(x[('L10', 0.1)], 20)
eq_(x.sub(source='L10').argmax(), 0.1)
eq_(x.sub(time=0.1).argmax(), 'L10')