def test_get_marked_results(self):
candidates = ([0.7, 0.2, 0.6, 0.8], 1.0)
params = (raptparams.Raptparams(), nccfparams.Nccfparams())
params[1].shortest_lag_per_frame = 7
params[0].min_acceptable_peak_val = 0.5
params[0].max_hypotheses_per_frame = 19
marked_values = pyrapt._get_marked_results(candidates, params, False)
self.assertEqual(3, len(marked_values))
self.assertEqual((9, 0.6), marked_values[1])
def test_get_marked_results_firstpass(self, mock_get_peak):
candidates = ([0.7, 0.2, 0.6, 0.8], 1.0)
params = (raptparams.Raptparams(), nccfparams.Nccfparams())
params[1].shortest_lag_per_frame = 7
params[0].min_acceptable_peak_val = 0.5
params[0].max_hypotheses_per_frame = 19
mock_get_peak.return_value = (9, 0.6)
marked_values = pyrapt._get_marked_results(candidates, params, True)
self.assertEqual(3, len(marked_values))
self.assertEqual((9, 0.6), marked_values[1])
mock_get_peak.assert_called_with(ANY, ANY, ANY)
def test_get_marked_results_above_max(self):
candidates = ([0.7, 0.2, 0.6, 0.8, 0.9, 0.5], 1.0)
params = (raptparams.Raptparams(), nccfparams.Nccfparams())
params[1].shortest_lag_per_frame = 7
params[0].min_acceptable_peak_val = 0.5
params[0].max_hypotheses_per_frame = 5
marked_values = pyrapt._get_marked_results(candidates, params, False)
self.assertEqual(4, len(marked_values))
self.assertEqual((7, 0.7), marked_values[0])
self.assertEqual((11, 0.9), marked_values[3])
self.assertEqual((10, 0.8), marked_values[2])
def test_get_results_for_frame(self, mock_get_for_all_lags):
mock_get_for_all_lags.return_value = ([0.2] * 35, 0.3)
audio = (2004, numpy.full(3346, 5.0))
params = (raptparams.Raptparams(), nccfparams.Nccfparams())
lag_range = 8
with patch('pyrapt.pyrapt._get_marked_results') as mock_mark:
mock_mark.return_value = [(9, 0.7), (15, 0.8), (17, 0.6)]
results = pyrapt._get_firstpass_frame_results(
audio, 5, lag_range, params)
mock_mark.assert_called_once_with(ANY, ANY, True)
self.assertEqual(3, len(results))
self.assertEqual(0.8, results[1][1])
def test_get_correlation(self):
audio = (10, numpy.array([0, 1, 2, 3, 4, 5, 6, 7]))
params = (raptparams.Raptparams(), nccfparams.Nccfparams())
params[1].samples_per_frame = 2
params[1].samples_correlated_per_lag = 5
correlation = pyrapt._get_correlation(audio, 0, 1, params)
self.assertEqual(0.816496580927726, correlation)
# Now try with additive constant added in denominator
# (only added for 2nd pass NCCF calc)
params[0].additive_constant = 12
correlation = pyrapt._get_correlation(audio, 0, 1, params, False)
self.assertEqual(0.7856742013183862, correlation)
def test_get_correlations_for_all_lags(self, mock_get_correlation):
mock_get_correlation.return_value = 0.4
audio = (2004, numpy.full(3346, 5.0))
params = (raptparams.Raptparams(), nccfparams.Nccfparams())
params[1].samples_correlated_per_lag = 20
params[1].samples_per_frame = 20
params[1].shortest_lag_per_frame = 10
lag_range = 8
results = pyrapt._get_correlations_for_all_lags(
audio, 5, lag_range, params)
self.assertEqual(0.4, results[1])
self.assertEqual(8, len(results[0]))
self.assertEqual(0.4, results[0][7])
mock_get_correlation.assert_called_with(ANY, ANY, ANY, ANY)
def test_get_second_pass_results_for_frame(self, mock_get_correlations):
with patch('pyrapt.pyrapt._get_marked_results') as mock_res:
mock_return_val = ([0.2] * 35, 0.5)
mock_get_correlations.return_value = mock_return_val
mock_res.return_value = [(4, 0.6), (4, 0.6), (4, 0.6)] * 165
audio = (44100, numpy.full(73612, 7.3))
i = 5
lag_range = 5
params = (raptparams.Raptparams(), nccfparams.Nccfparams())
params[0].sample_rate_ratio = 20
first_pass = [(4, 0.5), (5, 0.6), (22, 0.7)] * 165
frame_results = pyrapt._get_secondpass_frame_results(
audio, i, lag_range, params, first_pass)
mock_get_correlations.assert_called_once_with(
audio, 5, first_pass, 5, params)
mock_res.assert_called_once_with(mock_return_val, params, False)
self.assertEqual(4, frame_results[0][0])
def test_get_correlations_for_input_lags(self, mock_get_correlation):
mock_get_correlation.return_value = 0.6
audio = (44100, numpy.full(3500, 5.0))
params = (raptparams.Raptparams(), nccfparams.Nccfparams())
params[0].sample_rate_ratio = 4
params[1].samples_correlated_per_lag = 20
params[1].samples_per_frame = 100
params[1].shortest_lag_per_frame = 0
lag_range = 50
first_pass = [[(32, 0.7)]] * 35
results = pyrapt._get_correlations_for_input_lags(
audio, 5, first_pass, lag_range, params)
mock_get_correlation.assert_called_with(ANY, ANY, ANY, ANY, False)
self.assertEqual(50, len(results[0]))
self.assertEqual(0.6, results[0][32])
self.assertEqual(0.6, results[0][29])
self.assertEqual(0.6, results[0][35])
self.assertEqual(0.0, results[0][28])
self.assertEqual(0.0, results[0][36])
self.assertEqual(0.6, results[1])
def test_nccf_secondpass(self, mock_frame_results):
mock_frame_results.return_value = [(5, 0.6), (30, 0.7), (55, 0.9)]
first_pass = [(4, 0.6)] * 165
audio_data = (44100, numpy.full(73612, 6.8))
raptparam = raptparams.Raptparams()
raptparam.sample_rate_ratio = 20
with patch('pyrapt.pyrapt._get_nccf_params') as mock_get_params:
nccfparam = nccfparams.Nccfparams()
# lag range is supposed to be 0-samplerate/50 for 2nd pass default:
nccfparam.shortest_lag_per_frame = 0
nccfparam.longest_lag_per_frame = 882
nccfparam.max_frame_count = 165
params = (raptparam, nccfparam)
mock_get_params.return_value = nccfparam
candidates = pyrapt._second_pass_nccf(audio_data, first_pass,
raptparam)
mock_get_params.assert_called_once_with(audio_data, raptparam,
False)
# at each frame we get results, with lag range being max - 1
mock_frame_results.assert_called_with(audio_data, ANY, 881, params,
first_pass)
self.assertEqual(165, len(candidates))
self.assertEqual(5, candidates[0][0][0])