cryptorugmunch
0ecc250349
The duplicate file names (advanced, agency, auth, compliance, costing, etc.) in both root and routers/ made imports ambiguous. Renamed root modules to pry_<name>.py so: from quality import X -> from pry_quality import X from routers.quality import router (unchanged) Also: - Renamed x402.py to pry_x402/ package directory - Fixed 21+ bare imports across api.py, deps.py, routers/, tests/, llm_providers/ Tests: 593 passed, 1 skipped (test_ready_returns_200 fails pre-existing because Ollama is unreachable from test env, unrelated to this refactor). Audit item 8.
39 lines
1.1 KiB
Python
39 lines
1.1 KiB
Python
# SPDX-License-Identifier: MIT
|
|
# Copyright (c) 2026 Rug Munch Media LLC
|
|
#
|
|
# Part of Pry — https://git.rugmunch.io/RugMunchMedia/pryscraper
|
|
# Licensed under MIT. See LICENSE.
|
|
"""Tests for cost analytics engine."""
|
|
|
|
from pry_costing import (
|
|
DEFAULT_COST_TABLE,
|
|
get_monthly_usage,
|
|
record_usage,
|
|
)
|
|
|
|
|
|
def test_record_usage() -> None:
|
|
result = record_usage("scrape_direct", {"url": "https://example.com"})
|
|
assert result["operation"] == "scrape_direct"
|
|
assert result["cost"] > 0
|
|
assert "url" in result.get("metadata", {})
|
|
|
|
|
|
def test_record_usage_cost() -> None:
|
|
result = record_usage("llm_call", quantity=2)
|
|
expected = DEFAULT_COST_TABLE["llm_call"] * 2
|
|
assert result["cost"] == round(expected, 6)
|
|
|
|
|
|
def test_get_monthly_usage() -> None:
|
|
result = get_monthly_usage()
|
|
assert "total_cost" in result
|
|
assert "projected_monthly_cost" in result
|
|
assert "breakdown" in result
|
|
|
|
|
|
def test_cost_table_keys() -> None:
|
|
assert "scrape_direct" in DEFAULT_COST_TABLE
|
|
assert "llm_call" in DEFAULT_COST_TABLE
|
|
assert "vision_call" in DEFAULT_COST_TABLE
|
|
assert "storage_gb_month" in DEFAULT_COST_TABLE
|