def test_SumAllComponents(self):
factorization = SpleeterFactorization(self.dp, n_temporal_segments=1,
composition_fn=None,
model_name='spleeter:5stems',
spleeter_sources_path=spleeter_sources_path)
all_components = factorization.compose_model_input()
self.assertTrue(np.allclose(all_components, self.reference, atol=10**5))
def test_TemporalSegmentation(self):
n_segments = 7
factorization = SpleeterFactorization(self.dp, n_temporal_segments=n_segments,
composition_fn=None,
model_name='spleeter:5stems',
spleeter_sources_path=spleeter_sources_path)
all_components = factorization.compose_model_input()
leng = len(all_components) # to deal with ignored samples at the end
self.assertTrue(np.allclose(all_components, self.reference[:leng], atol=10 ** 5))
self.assertEqual(n_segments * 5, factorization.get_number_components()) # nr. sources = 5
def test_AnalysisWindow(self):
start = 35000
leng = 27333
reference = self.reference[start:start+leng]
factorization = SpleeterFactorization(self.dp, n_temporal_segments=1,
composition_fn=None,
model_name='spleeter:5stems',
spleeter_sources_path=spleeter_sources_path)
factorization.set_analysis_window(start, leng)
all_components = factorization.compose_model_input()
self.assertTrue(np.allclose(all_components, reference, atol=10 ** 5))