Testing the normal account

test_normal.py

import pytest
from decimal import Decimal
from main import Account


class TestNormalAccount:
    def test_add_amount(self) -> None:
        account = Account(
            amount=Decimal(10),
            blocked=Decimal(0),
        )
        account.add_amount(Decimal(5))
        assert account.available_amount == 15

    def test_block_amount(self) -> None:
        account = Account(
            amount=Decimal(10),
            blocked=Decimal(0),
        )
        with pytest.raises(ValueError):
            account.block_amount(Decimal(11))
        assert account.available_amount == 10
        account.block_amount(Decimal(7))
        assert account.available_amount == 3