def test_spectrum(self):
# test 2d spectrum
spectrum = spade.concepts_mining(self.patt1, self.binsize,
self.winlen, report='#')[0]
# test 3d spectrum
assert_array_equal(spectrum, [[len(self.lags1) + 1, self.n_occ1, 1]])
spectrum_3d = spade.concepts_mining(self.patt1, self.binsize,
self.winlen, report='3d#')[0]
assert_array_equal(spectrum_3d, [
[len(self.lags1) + 1, self.n_occ1, max(self.lags1), 1]])
def test_spade_cpp(self):
# Computing Spectrum
spectrum_cpp = spade.concepts_mining(self.cpp,
self.binsize,
1,
report='#')[0]
# Check spectrum
assert_array_equal(spectrum_cpp,
[(len(self.cpp), len(self.cpp[0]), 1)])
def test_spade_spectrum_cpp(self):
# Computing Spectrum
spectrum_cpp = spade.concepts_mining(self.cpp, self.bin_size,
1, report='#')[0]
# Check spectrum
assert_array_equal(
spectrum_cpp,
[(len(self.cpp),
np.sum(conv.BinnedSpikeTrain(
self.cpp[0], self.bin_size).to_bool_array()), 1)])
