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)])