test: fix missing tests

tests/datasets/test_datasets_schemas.py

@@ -1,6 +1,7 @@
 import pandas as pd
 import pytest
 
+from siapy.core.exceptions import InvalidInputError
 from siapy.datasets.schemas import (
     ClassificationTarget,
     RegressionTarget,
@@ -270,7 +271,7 @@ def test_tabular_dataset_data_target_from_dict_classification():
 
 def test_tabular_dataset_data_target_from_dict_invalid():
     data = {"unknown_key": "some_value"}
-    with pytest.raises(ValueError):
+    with pytest.raises(InvalidInputError):
         TabularDatasetData.target_from_dict(data)
 
 
@@ -341,15 +342,17 @@ def test_tabular_dataset_data_setattr():
     # Invalid setattr -> signals
     invalid_signals = pd.DataFrame({"b": [4, 5]})
     with pytest.raises(
-        ValueError, match="Lengths of pixels, signals, and metadata must be equal"
+        InvalidInputError,
+        match="Lengths of pixels, signals, and metadata must be equal",
     ):
         dataset.signals = invalid_signals
 
     # Invalid setattr -> target
     dataset = dataset_.model_copy()
     invalid_target = target[:2]
     with pytest.raises(
-        ValueError, match="Target length must be equal to the length of the dataset."
+        InvalidInputError,
+        match="Target length must be equal to the length of the dataset.",
     ):
         dataset.target = invalid_target