第 3 章：Claude Code 高效工作流

RED 阶段：AI 生成测试

import pytest
from disk_utils import DiskCalculator, DiskInfo

class TestDiskCalculator:
    def test_parse_size_bytes(self):
        """解析纯字节数"""
        assert DiskCalculator.parse_size("1024") == 1024

    def test_parse_size_kb(self):
        """解析 KB 单位"""
        assert DiskCalculator.parse_size("1K") == 1024

    @pytest.mark.parametrize("size_str,expected", [
        ("1.5G", int(1.5 * 1024**3)),
        ("2T", 2 * 1024**4),
        ("100M", 100 * 1024**2),
    ])
    def test_parse_size_various_units(self, size_str, expected):
        assert DiskCalculator.parse_size(size_str) == expected

    def test_format_size_bytes(self):
        """小文件使用字节显示"""
        assert DiskCalculator.format_size(500) == "500 B"

    def test_format_size_auto_unit(self):
        """自动选择合适的单位"""
        assert DiskCalculator.format_size(2 * 1024**3) == "2.00 GiB"

    def test_parse_invalid_format_raises(self):
        """非法格式抛出 ValueError"""
        with pytest.raises(ValueError, match="无法解析"):
            DiskCalculator.parse_size("not_a_size")

    def test_calculate_usage_percent(self):
        """计算磁盘使用率"""
        assert DiskCalculator.usage_percent(used=50, total=100) == 50.0

    def test_usage_percent_zero_total(self):
        """总量为 0 时返回 0 避免除零错误"""
        assert DiskCalculator.usage_percent(used=0, total=0) == 0.0

GREEN 阶段：AI 实现代码

"""磁盘工具模块 — 磁盘空间计算与格式化。"""

import re
from dataclasses import dataclass


@dataclass
class DiskInfo:
    """磁盘信息数据类。"""
    device: str
    mount_point: str
    total_bytes: int
    used_bytes: int
    filesystem: str


class DiskCalculator:
    """磁盘空间计算工具。"""

    _SIZE_UNITS = {
        'B': 1, 'K': 1024, 'M': 1024**2,
        'G': 1024**3, 'T': 1024**4, 'P': 1024**5,
    }

    _SIZE_PATTERN = re.compile(
        r'^([\d.]+)\s*([BKMGTP])?$', re.IGNORECASE
    )

    @classmethod
    def parse_size(cls, size_str: str) -> int:
        """解析人类可读的大小字符串为字节数。

        Args:
            size_str: 如 "1.5G", "100M", "1024"

        Returns:
            字节数（整数）

        Raises:
            ValueError: 如果格式无法解析
        """
        match = cls._SIZE_PATTERN.match(size_str.strip())
        if not match:
            raise ValueError(f"无法解析大小字符串: '{size_str}'")
        value = float(match.group(1))
        unit = (match.group(2) or 'B').upper()
        return int(value * cls._SIZE_UNITS.get(unit, 1))

    @classmethod
    def format_size(cls, bytes_val: int) -> str:
        """将字节数格式化为人类可读的字符串。

        Args:
            bytes_val: 字节数

        Returns:
            格式化字符串，如 "2.50 GiB"
        """
        if bytes_val < 1024:
            return f"{bytes_val} B"
        for unit in ['KiB', 'MiB', 'GiB', 'TiB', 'PiB']:
            bytes_val /= 1024
            if bytes_val < 1024:
                return f"{bytes_val:.2f} {unit}"
        return f"{bytes_val:.2f} PiB"

    @staticmethod
    def usage_percent(used: int, total: int) -> float:
        """计算使用百分比，安全处理除零。

        Args:
            used: 已使用字节数
            total: 总字节数

        Returns:
            使用百分比 (0.0 - 100.0)
        """
        if total == 0:
            return 0.0
        return round((used / total) * 100, 2)

REFACTOR 阶段：优化建议