"""Test the functioning of loading .asd files."""

from pathlib import Path

import pytest

from AFMReader import asd

BASE_DIR = Path.cwd()
RESOURCES = BASE_DIR / "tests" / "resources"


@pytest.mark.parametrize(
    ("file_name", "channel", "number_of_frames", "pixel_to_nm_scaling"),
    [
        pytest.param("sample_0.asd", "TP", 142, 0.78125, id="file type 0"),
        pytest.param("sample_1.asd", "TP", 197, 2.0, id="file type 1"),
    ],
)
def test_load_asd(file_name: str, channel: str, number_of_frames: int, pixel_to_nm_scaling: float) -> None:
    """Test the normal operation of loading a .asd file."""
    result_frames = list
    result_pixel_to_nm_scaling = float
    result_metadata = dict

    file_path = RESOURCES / file_name
    result_frames, result_pixel_to_nm_scaling, result_metadata = asd.load_asd(file_path, channel)

    assert len(result_frames) == number_of_frames  # type: ignore
    assert result_pixel_to_nm_scaling == pixel_to_nm_scaling
    assert isinstance(result_metadata, dict)


def test_load_asd_file_not_found() -> None:
    """Ensure FileNotFound error is raised."""
    with pytest.raises(FileNotFoundError):
        asd.load_asd("nonexistant_file.asd", channel="TP")