def test_labels_to_indices(): """Test if the label to index converter works as expected.""" dataloader = DummyLoader(classes=[ 'mood/theme---a', 'mood/theme---b', 'mood/theme---c', 'mood/theme---d', 'mood/theme---e', 'mood/theme---f', 'mood/theme---g', 'mood/theme---h', ]) label_list = ['b', 'e', 'mood/theme---f', 'a', 'mood/theme---c'] expected = [1, 4, 5, 0, 2] actual = labels_to_indices(dataloader=dataloader, label_list=label_list) assert (expected == actual).all()
labels_to_indices( dataloader=dataloader, label_list=[ # theme 'action', 'adventure', 'advertising', 'background', 'ballad', 'children', 'christmas', 'commercial', 'corporate', 'documentary', 'drama', 'dream', 'film', 'game', 'holiday', 'love', 'movie', 'nature', 'party', 'retro', 'soundscape', 'space', 'sport', 'summer', 'trailer', 'travel', ], ),