mirror of
https://github.com/BerriAI/litellm.git
synced 2026-10-10 03:28:53 +00:00
test(e2e): type the wire-shape parametrize ids callback
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
parent
feb69c5f78
commit
5507de326e
1 changed files with 5 additions and 1 deletions
|
|
@ -144,6 +144,10 @@ _SHAPES: Final[tuple[tuple[str, str, Case], ...]] = (
|
|||
)
|
||||
|
||||
|
||||
def _shape_id(entry: tuple[str, str, Case]) -> str:
|
||||
return entry[0]
|
||||
|
||||
|
||||
class TestWireFormats:
|
||||
@pytest.mark.parametrize("wire", tuple(_WIRE_USAGE))
|
||||
@pytest.mark.covers("quota_management.spend_tracking.scripted_wire.logs_cost")
|
||||
|
|
@ -237,7 +241,7 @@ class TestWireFormats:
|
|||
)
|
||||
cost_rows.assert_total_is_sum_of_components(row)
|
||||
|
||||
@pytest.mark.parametrize("shape_wire_case", _SHAPES, ids=lambda entry: entry[0])
|
||||
@pytest.mark.parametrize("shape_wire_case", _SHAPES, ids=_shape_id)
|
||||
@pytest.mark.covers("quota_management.spend_tracking.scripted_wire.logs_cost")
|
||||
def test_response_shape_bills_reported_usage(
|
||||
self,
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue