andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-01-09 16:18:30 +00:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

fix: rename test classes to avoid pytest collection warnings

Browse Source 
- Renamed TestEvent to SampleEvent in test_crewai_event_bus.py
- Renamed TestModel to SampleModel in test_training_converter.py
This commit is contained in:
Greyson LaLonde
2025-09-19 00:18:43 -04:00
parent 080a7d753f
commit 2607f5361c
2 changed files with 7 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
packages/crewai/tests/utilities/events/test_crewai_event_bus.py
View File

@@ -4,18 +4,18 @@ from crewai.events.base_events import BaseEvent
from crewai.events.event_bus import crewai_event_bus from crewai.events.event_bus import crewai_event_bus
class TestEvent(BaseEvent): class SampleEvent(BaseEvent):
pass pass
def test_specific_event_handler(): def test_specific_event_handler():
mock_handler = Mock() mock_handler = Mock()
@crewai_event_bus.on(TestEvent) @crewai_event_bus.on(SampleEvent)
def handler(source, event): def handler(source, event):
mock_handler(source, event) mock_handler(source, event)
event = TestEvent(type="test_event") event = SampleEvent(type="test_event")
crewai_event_bus.emit("source_object", event) crewai_event_bus.emit("source_object", event)
mock_handler.assert_called_once_with("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): def handler(source, event):
mock_handler(source, event) mock_handler(source, event)
event = TestEvent(type="test_event") event = SampleEvent(type="test_event")
crewai_event_bus.emit("source_object", event) crewai_event_bus.emit("source_object", event)
mock_handler.assert_called_once_with("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): def broken_handler(source, event):
raise ValueError("Simulated handler failure") raise ValueError("Simulated handler failure")
event = TestEvent(type="test_event") event = SampleEvent(type="test_event")
crewai_event_bus.emit("source_object", event) crewai_event_bus.emit("source_object", event)
out, err = capfd.readouterr() out, err = capfd.readouterr()

4
packages/crewai/tests/utilities/test_training_converter.py
View File

@@ -7,7 +7,7 @@ from crewai.utilities.converter import ConverterError
from crewai.utilities.training_converter import TrainingConverter from crewai.utilities.training_converter import TrainingConverter
class TestModel(BaseModel): class SampleModel(BaseModel):
string_field: str = Field(description="A simple string field") string_field: str = Field(description="A simple string field")
list_field: List[str] = Field(description="A list of strings") list_field: List[str] = Field(description="A list of strings")
number_field: float = Field(description="A number field") number_field: float = Field(description="A number field")
@@ -22,7 +22,7 @@ class TestTrainingConverter:
self.converter = TrainingConverter( self.converter = TrainingConverter(
llm=self.llm_mock, llm=self.llm_mock,
text=self.test_text, text=self.test_text,
model=TestModel, model=SampleModel,
instructions=self.test_instructions instructions=self.test_instructions
) )