tests

Author: hlky

tests/pipelines/test_pipelines.py

@@ -1811,31 +1811,34 @@ def test_wrong_model(self):
                 "hf-internal-testing/diffusers-stable-diffusion-tiny-all", text_encoder=tokenizer
             )
 
-        assert "Expected" in str(error_context.exception)
-        assert "text_encoder" in str(error_context.exception)
-        assert "CLIPTokenizer" in str(error_context.exception)
+        assert "is of type" in str(error_context.exception)
+        assert "but should be" in str(error_context.exception)
 
     def test_wrong_model_scheduler_type(self):
         scheduler = EulerDiscreteScheduler.from_pretrained("hf-internal-testing/tiny-flux-pipe", subfolder="scheduler")
-        with self.assertRaises(ValueError) as error_context:
+        with self.assertLogs(
+            logging.get_logger("diffusers.pipelines.pipeline_utils"), level="WARNING"
+        ) as warning_context:
             _ = FluxPipeline.from_pretrained("hf-internal-testing/tiny-flux-pipe", scheduler=scheduler)
 
-        assert "Expected" in str(error_context.exception)
-        assert "scheduler" in str(error_context.exception)
-        assert "EulerDiscreteScheduler" in str(error_context.exception)
+        assert any("Expected" in message for message in warning_context.output)
+        assert any("scheduler" in message for message in warning_context.output)
+        assert any("EulerDiscreteScheduler" in message for message in warning_context.output)
 
     def test_wrong_model_scheduler_enum(self):
         scheduler = FlowMatchEulerDiscreteScheduler.from_pretrained(
             "hf-internal-testing/diffusers-stable-diffusion-tiny-all", subfolder="scheduler"
         )
-        with self.assertRaises(ValueError) as error_context:
+        with self.assertLogs(
+            logging.get_logger("diffusers.pipelines.pipeline_utils"), level="WARNING"
+        ) as warning_context:
             _ = StableDiffusionPipeline.from_pretrained(
                 "hf-internal-testing/diffusers-stable-diffusion-tiny-all", scheduler=scheduler
             )
 
-        assert "Expected" in str(error_context.exception)
-        assert "scheduler" in str(error_context.exception)
-        assert "FlowMatchEulerDiscreteScheduler" in str(error_context.exception)
+        assert any("Expected" in message for message in warning_context.output)
+        assert any("scheduler" in message for message in warning_context.output)
+        assert any("FlowMatchEulerDiscreteScheduler" in message for message in warning_context.output)
 
 
 @slow