diff --git a/packages/crewai/tests/utilities/events/test_crewai_event_bus.py b/packages/crewai/tests/utilities/events/test_crewai_event_bus.py index f034de595..3cc44a408 100644 --- a/packages/crewai/tests/utilities/events/test_crewai_event_bus.py +++ b/packages/crewai/tests/utilities/events/test_crewai_event_bus.py @@ -4,18 +4,18 @@ from crewai.events.base_events import BaseEvent from crewai.events.event_bus import crewai_event_bus -class TestEvent(BaseEvent): +class SampleEvent(BaseEvent): pass def test_specific_event_handler(): mock_handler = Mock() - @crewai_event_bus.on(TestEvent) + @crewai_event_bus.on(SampleEvent) def handler(source, event): mock_handler(source, event) - event = TestEvent(type="test_event") + event = SampleEvent(type="test_event") crewai_event_bus.emit("source_object", event) mock_handler.assert_called_once_with("source_object", event) @@ -28,7 +28,7 @@ def test_wildcard_event_handler(): def handler(source, event): mock_handler(source, event) - event = TestEvent(type="test_event") + event = SampleEvent(type="test_event") crewai_event_bus.emit("source_object", event) mock_handler.assert_called_once_with("source_object", event) @@ -39,7 +39,7 @@ def test_event_bus_error_handling(capfd): def broken_handler(source, event): raise ValueError("Simulated handler failure") - event = TestEvent(type="test_event") + event = SampleEvent(type="test_event") crewai_event_bus.emit("source_object", event) out, err = capfd.readouterr() diff --git a/packages/crewai/tests/utilities/test_training_converter.py b/packages/crewai/tests/utilities/test_training_converter.py index 7eb21ae81..8bfaa0e82 100644 --- a/packages/crewai/tests/utilities/test_training_converter.py +++ b/packages/crewai/tests/utilities/test_training_converter.py @@ -7,7 +7,7 @@ from crewai.utilities.converter import ConverterError from crewai.utilities.training_converter import TrainingConverter -class TestModel(BaseModel): +class SampleModel(BaseModel): string_field: str = Field(description="A simple string field") list_field: List[str] = Field(description="A list of strings") number_field: float = Field(description="A number field") @@ -22,7 +22,7 @@ class TestTrainingConverter: self.converter = TrainingConverter( llm=self.llm_mock, text=self.test_text, - model=TestModel, + model=SampleModel, instructions=self.test_instructions )