def test_indices2words(self): v = Vocab(unk='unk') v.update(['i', 'like', 'pie']) words = v.indices2words([1, 2, 3, 0]) assert words == ['i', 'like', 'pie', 'unk']