def _test_bg_subtraction2(self):
p = Preprocessor(10)
s = p.load_npy('./test.npy')
generator = DataPreparator("", "", 512)
samples1=len(s[0,:])
snew,sbg = generator.bg_subtraction(s)
samples2=len(snew[0,:])
self.assertGreater(samples1, samples2)
def _test_bg_subtraction(self):
p=Preprocessor(10)
s=p.load_npy('./test.npy')
generator = DataPreparator("", "", 512)
mask, vector = generator.create_mask(s)
self.assertGreater(np.sum(vector), 120)
