()
| 46 | |
| 47 | |
| 48 | def test_textdecoder_default(): |
| 49 | chars_dict = {"<PAD>": 0, "<EOS>": 1, "A": 2, "B": 3, "C": 4} |
| 50 | decoder = TextDecoder(chars_dict=chars_dict, decode_mode=DecodeMode.Normal) |
| 51 | row = np.array([2, 3, 4, 1, 0], dtype=np.int32) |
| 52 | result = decoder.decode([row]) |
| 53 | assert result == ["ABC"] |
| 54 | |
| 55 | |
| 56 | def test_textdecoder_no_eos(): |
nothing calls this directly
no test coverage detected