test: Extended unittest

mindee · Nov 17, 2021 · bbf5676 · bbf5676
1 parent bda3773
commit bbf5676
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/common/test_datasets_utils.py b/tests/common/test_datasets_utils.py
@@ -47,6 +47,8 @@ def test_decode_sequence():
     with pytest.raises(AssertionError):
         utils.decode_sequence(np.array([2, 4.5]), mapping)
 
+    assert utils.decode_sequence([3, 4, 3, 4], mapping) == "dede"
+
 
 @pytest.mark.parametrize(
     "sequences, vocab, target_size, sos, eos, pad, dynamic_len, error, out_shape, gts",