Test Generator Pack

from test_generator_pack.tool import run

result = run(
    source_code="""
def calculate_discount(price: float, tier: str) -> float:
    rates = {"bronze": 0.05, "silver": 0.10, "gold": 0.15, "platinum": 0.20}
    if price < 0:
        raise ValueError("Price cannot be negative")
    rate = rates.get(tier, 0.0)
    return round(price * (1 - rate), 2)
""",
    framework="pytest",
    include_edge_cases=True
)

print(result["test_code"])
# import pytest
# from module import calculate_discount
#
# def test_gold_tier_discount():
#     assert calculate_discount(100.00, "gold") == 85.00
#
# def test_unknown_tier_no_discount():
#     assert calculate_discount(50.00, "unknown") == 50.00
#
# def test_negative_price_raises():
#     with pytest.raises(ValueError):
#         calculate_discount(-10.00, "silver")
#
# def test_zero_price():
#     assert calculate_discount(0.0, "platinum") == 0.0

print(f"Generated {result['test_count']} tests")
print(f"Coverage targets: {result['coverage_targets']}")