add missing image test, dont know why it went missing

Safe-DS · Jul 13, 2024 · 7ff98d2 · 7ff98d2
1 parent 02c0f05
commit 7ff98d2
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/tests/safeds/ml/nn/test_model.py b/tests/safeds/ml/nn/test_model.py
@@ -567,6 +567,11 @@ def test_should_raise_if_test_features_mismatch(self, device: Device) -> None:
                 [FlattenLayer()],
                 r"A NeuralNetworkClassifier cannot be used with a InputConversionImage that uses a VariableImageSize.",
             ),
+            (
+                InputConversionImageToImage(VariableImageSize(1, 1, 1)),
+                [FlattenLayer()],
+                r"A NeuralNetworkClassifier cannot be used with images as output.",
+            ),
         ],
     )
     def test_should_raise_if_model_has_invalid_structure(