diff --git a/tests/unit/test_streaming_billing_finalization.py b/tests/unit/test_streaming_billing_finalization.py index be9ff3fb..22ffc7a3 100644 --- a/tests/unit/test_streaming_billing_finalization.py +++ b/tests/unit/test_streaming_billing_finalization.py @@ -3,12 +3,11 @@ from unittest.mock import AsyncMock, MagicMock, patch import pytest from sqlalchemy.exc import SQLAlchemyError - -import routstr.auth as auth_module from sqlalchemy.ext.asyncio import AsyncEngine, create_async_engine from sqlmodel import SQLModel from sqlmodel.ext.asyncio.session import AsyncSession +import routstr.auth as auth_module from routstr.auth import ( adjust_payment_for_tokens, get_reservation_snapshot,