import pytest

from usage import BalanceResult


def test_balance_result_dataclass():
    result = BalanceResult(balance=10.5, is_invalid=False, error=None)
    assert result.balance == 10.5
    assert result.is_invalid is False
    assert result.error is None


def test_balance_result_invalid():
    result = BalanceResult(balance=None, is_invalid=True, error="Invalid token")
    assert result.balance is None
    assert result.is_invalid is True
    assert result.error == "Invalid token"