testsAndMisc/articles/test_site_size.py

"""Tests to ensure website stays within size budget."""

import os

# Budget for the entire website (single file) in bytes
BUDGET = 14 * 1024  # 14 KiB

HERE = os.path.dirname(__file__)
SITE_FILE = os.path.join(HERE, "index.html")


def test_site_file_exists() -> None:
    """Verify the main site HTML file exists."""
    assert os.path.exists(SITE_FILE), f"Missing site file: {SITE_FILE}"


def test_site_size_under_budget() -> None:
    """Verify site size is under the defined budget."""
    size = os.path.getsize(SITE_FILE)
    assert size <= BUDGET, f"Site size {size} bytes exceeds budget {BUDGET}"