mirror of
https://github.com/tiennm99/litellm.git
synced 2026-07-13 13:10:18 +00:00
Ishaan Jaffer
48 lines
1.5 KiB
Python
48 lines
1.5 KiB
Python
"""Tests for FocusCsvSerializer."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import polars as pl
|
|
|
|
from litellm.integrations.focus.serializers.csv import FocusCsvSerializer
|
|
|
|
|
|
def test_should_serialize_dataframe_to_csv():
|
|
frame = pl.DataFrame(
|
|
{"BilledCost": [1.5, 2.0], "ServiceName": ["openai", "anthropic"]}
|
|
)
|
|
serializer = FocusCsvSerializer()
|
|
result = serializer.serialize(frame)
|
|
|
|
assert isinstance(result, bytes)
|
|
lines = result.decode("utf-8").strip().split("\n")
|
|
assert lines[0] == "BilledCost,ServiceName"
|
|
assert len(lines) == 3 # header + 2 data rows
|
|
|
|
|
|
def test_should_return_header_only_for_empty_frame():
|
|
frame = pl.DataFrame(schema={"BilledCost": pl.Float64, "ServiceName": pl.Utf8})
|
|
serializer = FocusCsvSerializer()
|
|
result = serializer.serialize(frame)
|
|
|
|
lines = result.decode("utf-8").strip().split("\n")
|
|
assert lines[0] == "BilledCost,ServiceName"
|
|
assert len(lines) == 1 # header only
|
|
|
|
|
|
def test_should_cast_decimal_columns_to_float():
|
|
frame = pl.DataFrame(
|
|
{"BilledCost": [1, 2], "ServiceName": ["openai", "anthropic"]}
|
|
).cast({"BilledCost": pl.Decimal(18, 6)})
|
|
serializer = FocusCsvSerializer()
|
|
result = serializer.serialize(frame)
|
|
|
|
lines = result.decode("utf-8").strip().split("\n")
|
|
# Should use floating-point notation (e.g. "1.0") not fixed-point ("1.000000")
|
|
assert "1.000000" not in lines[1]
|
|
assert lines[1].startswith("1.0,")
|
|
|
|
|
|
def test_extension_should_be_csv():
|
|
assert FocusCsvSerializer.extension == "csv"
|