added test

Author: philschmid

tests/unit/test_decoder_encoder.py

@@ -12,8 +12,9 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 import json
-
+import pytest
 from sagemaker_huggingface_inference_toolkit import decoder_encoder
+from mms.service import PredictionException
 
 
 ENCODE_JSON_INPUT = {"upper": [1425], "lower": [576], "level": [2], "datetime": ["2012-08-08 15:30"]}
@@ -46,6 +47,13 @@ def test_decode_csv():
     assert decoded_data == {"inputs": ["I love you", "I like you"]}
 
 
+def test_decode_csv_without_header():
+    with pytest.raises(PredictionException):
+        decoder_encoder.decode_csv(
+            "where do i live?,My name is Philipp and I live in Nuremberg\r\nwhere is Berlin?,Berlin is the capital of Germany"
+        )
+
+
 def test_encode_json():
     encoded_data = decoder_encoder.encode_json(ENCODE_JSON_INPUT)
     assert json.loads(encoded_data) == ENCODE_JSON_INPUT