Spaces:
Sleeping
Sleeping
File size: 1,525 Bytes
b610d23 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 |
"""Tests for core data models."""
import pytest
from datetime import datetime
from pathlib import Path
from src.models import ImageMetadata, DownloadTask, PlaybackState, TaskStatus
def test_image_metadata_creation():
"""Test ImageMetadata dataclass creation."""
now = datetime.now()
metadata = ImageMetadata(
date=now,
time_sequence="120000",
filename="20251219_120000_4096_0211.jpg",
local_path=Path("data/2025/12/19/20251219_120000_4096_0211.jpg"),
file_size=1024000,
download_timestamp=now,
url="https://sdo.gsfc.nasa.gov/assets/img/browse/2025/12/19/20251219_120000_4096_0211.jpg"
)
assert metadata.date == now
assert metadata.time_sequence == "120000"
assert metadata.filename == "20251219_120000_4096_0211.jpg"
assert metadata.file_size == 1024000
def test_download_task_defaults():
"""Test DownloadTask default values."""
task = DownloadTask(
url="https://example.com/image.jpg",
target_path=Path("data/image.jpg")
)
assert task.retry_count == 0
assert task.status == TaskStatus.PENDING
assert task.error_message is None
def test_playback_state_defaults():
"""Test PlaybackState default values."""
state = PlaybackState()
assert state.current_frame == 0
assert state.total_frames == 0
assert state.is_playing is False
assert state.playback_speed == 1.0
assert state.selected_date_range is None |