diff --git a/scripts/test_nfo_integration.py b/scripts/test_nfo_integration.py index 81ec0bf..6c1bae2 100644 --- a/scripts/test_nfo_integration.py +++ b/scripts/test_nfo_integration.py @@ -22,9 +22,9 @@ from pathlib import Path # Add src to path sys.path.insert(0, str(Path(__file__).parent.parent)) -from src.core.services.tmdb_client import TMDBClient, TMDBAPIError -from src.core.services.nfo_service import NFOService from src.core.entities.nfo_models import TVShowNFO +from src.core.services.nfo_service import NFOService +from src.core.services.tmdb_client import TMDBAPIError, TMDBClient from src.core.utils.nfo_generator import generate_tvshow_nfo, validate_nfo_xml @@ -99,7 +99,12 @@ async def test_nfo_generation(): try: # Create a sample NFO model print("\n1. Creating sample TVShowNFO model...") - from src.core.entities.nfo_models import RatingInfo, ActorInfo, ImageInfo, UniqueID + from src.core.entities.nfo_models import ( + ActorInfo, + ImageInfo, + RatingInfo, + UniqueID, + ) nfo = TVShowNFO( title="Test Show",