def test_get_leaves2(self):
mtrack = multitrack.MultiTrack('NightPanther_Fire')
test_dict = {
'a': mtrack,
'b': {1: mtrack, 2: mtrack},
'c': [mtrack],
'd': {'asdf': mtrack},
'e': {'borg': [mtrack]}
}
actual = multitrack.get_dict_leaves(test_dict)
expected = set([mtrack, mtrack, mtrack, mtrack, mtrack])
self.assertEqual(actual, expected)
def test_defaults_no_mixing_coeffs(self):
mtrack = multitrack.MultiTrack('AHa_TakeOnMe')
actual_filepaths, actual_weights = mix._build_mix_args(
mtrack, None, None, None, None)
expected_filepaths = [
os.path.join(AUDIO_PATH, 'AHa_TakeOnMe', 'AHa_TakeOnMe_STEMS',
'AHa_TakeOnMe_STEM_{}.wav'.format(i))
for i in ['01', '02', '03', '04', '05', '06']
]
expected_weights = [1, 1, 1, 1, 1, 1]
self.assertEqual(expected_filepaths, actual_filepaths)
self.assertEqual(expected_weights, actual_weights)
def setUp(self):
self.mtrack = multitrack.MultiTrack('LizNelson_Rainfall')
def setUp(self):
self.mtrack = multitrack.MultiTrack('Phoenix_ScotchMorris')
def setUp(self):
self.mtrack = multitrack.MultiTrack("NightPanther_Fire")
self.mtrack2 = multitrack.MultiTrack("Phoenix_ScotchMorris")
self.stem = self.mtrack.stems[8]
self.raw = self.mtrack.raw_audio[8][1]
def test_invalid_trackid(self):
with self.assertRaises(IOError):
multitrack.MultiTrack("RickAstley_NeverGonnaGiveYouUp")
def test_get_mixing_coefficient(self):
mtrack = multitrack.MultiTrack('AClassicEducation_NightOwl')
actual = mtrack._get_mixing_coefficient(3)
expected = 0.2
self.assertEqual(actual, expected)
def test_dataset_version_extra(self):
mtrack = multitrack.MultiTrack("AHa_TakeOnMe")
actual = mtrack.dataset_version
expected = 'EXTRA'
self.assertEqual(expected, actual)
def test_dataset_version_v2(self):
mtrack = multitrack.MultiTrack("FennelCartwright_DearTessie")
actual = mtrack.dataset_version
expected = 'V2'
self.assertEqual(expected, actual)
