17 lines
493 B
Python
17 lines
493 B
Python
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"
|