diff --git a/tests/verification/test_cross_node_blockchain.py b/tests/verification/test_cross_node_blockchain.py index e491b36c..1279b306 100644 --- a/tests/verification/test_cross_node_blockchain.py +++ b/tests/verification/test_cross_node_blockchain.py @@ -8,7 +8,7 @@ import hashlib import subprocess from datetime import datetime import time -from aitbc import AITBCHTTPClient, NetworkErroration +from aitbc import AITBCHTTPClient, NetworkError # Test configuration NODES = { diff --git a/tests/verification/test_model_validation.py b/tests/verification/test_model_validation.py index 3770e653..e593a8b2 100644 --- a/tests/verification/test_model_validation.py +++ b/tests/verification/test_model_validation.py @@ -5,6 +5,7 @@ Test the BlockImportRequest model from pydantic import BaseModel, Field from typing import Dict, Any, List, Optional +from aitbc import validate_address, validate_hash class TransactionData(BaseModel): tx_hash: str diff --git a/tests/verification/test_payment_integration.py b/tests/verification/test_payment_integration.py index 6b893fca..69ea5f30 100755 --- a/tests/verification/test_payment_integration.py +++ b/tests/verification/test_payment_integration.py @@ -5,11 +5,10 @@ Tests job creation with payments, escrow, release, and refund flows """ import asyncio -import httpx import json from datetime import datetime from typing import Dict, Any -from aitbc import get_logger +from aitbc import get_logger, AITBCHTTPClient # Configure logging logger = get_logger(__name__)