tests/conftest.py

"""Pytest configuration and fixtures"""

import os
import pathlib
import tempfile

import pytest


@pytest.fixture
def temp_dir():
    with tempfile.TemporaryDirectory() as tmp_dir:
        yield pathlib.Path(tmp_dir)


@pytest.fixture
def test_gguf_url():
    return "https://huggingface.co/bartowski/SmolLM2-135M-Instruct-GGUF/resolve/main/SmolLM2-135M-Instruct-Q4_K_M.gguf"


@pytest.fixture
def test_repo_id():
    return "bartowski/SmolLM2-135M-Instruct-GGUF"


@pytest.fixture
def test_gguf_filename():
    return "SmolLM2-135M-Instruct-Q4_K_M.gguf"