def test_imfs_and_residue_accessor(self): S = np.random.random(100) eemd = EEMD(trials=5, max_imf=2, parallel=False) eIMFs = eemd(S) imfs, residue = eemd.get_imfs_and_residue() self.assertEqual(eIMFs.shape[0], imfs.shape[0], "Compare number of components") self.assertEqual(len(residue), 100, "Check if residue exists")
def extract_imfs(self, signal): eemd = EEMD() eemd.eemd(signal) res = eemd.get_imfs_and_residue() imfs = list(res)[0] # drop the noise/ high frequency imfs, may incur index error due to the signal is too simple imfs_left = imfs[-1:] return imfs_left # np.array type
def test_imfs_and_residue_accessor2(self): eemd = EEMD() with self.assertRaises(ValueError): imfs, residue = eemd.get_imfs_and_residue()