mirror of
https://github.com/BerriAI/litellm.git
synced 2026-10-11 03:38:38 +00:00
36 lines
1.3 KiB
Python
36 lines
1.3 KiB
Python
from typing import Final
|
|
|
|
from tests.integration.run import select, uncollected
|
|
|
|
_GROUP: Final = (
|
|
"tests/integration/cost_calculation/test_cost_tracking.py",
|
|
"tests/integration/cost_calculation/test_rollups.py",
|
|
)
|
|
_CELL: Final = (
|
|
"tests/integration/cost_calculation/test_cost_tracking.py"
|
|
"::test_case_bills_expected_cost[perplexity/pplx-decider-v1-27b-decisions]"
|
|
)
|
|
|
|
|
|
def test_a_node_id_inside_a_group_file_is_selected_as_written() -> None:
|
|
selection: Final = select((_CELL,), _GROUP)
|
|
assert selection.nodes == (_CELL,)
|
|
assert selection.foreign == ()
|
|
|
|
|
|
def test_a_node_id_outside_the_group_is_foreign_by_its_file() -> None:
|
|
foreign: Final = "tests/integration/providers/test_decisions_wire.py::test_key_checks_match_chat"
|
|
assert select((foreign, _CELL), _GROUP).foreign == (foreign,)
|
|
|
|
|
|
def test_no_request_selects_every_group_file() -> None:
|
|
assert select((), _GROUP).nodes == _GROUP
|
|
|
|
|
|
def test_a_node_id_whose_file_collected_tests_is_not_empty() -> None:
|
|
collected: Final = frozenset({_CELL, "tests/integration/cost_calculation/test_cost_tracking.py::test_other"})
|
|
assert uncollected((_CELL,), collected) == ()
|
|
|
|
|
|
def test_a_selected_file_that_collected_nothing_is_reported() -> None:
|
|
assert uncollected(_GROUP, frozenset({_CELL})) == ("tests/integration/cost_calculation/test_rollups.py",)
|