andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-01-08 23:58:34 +00:00
Code Issues Actions 4 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
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()

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
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
)