31 lines
741 B
Python
31 lines
741 B
Python
from pathlib import Path
|
|
from typing import Generator, Callable
|
|
|
|
import pytest
|
|
|
|
from intake.source import LocalSource
|
|
|
|
|
|
def clean_source(source_path: Path):
|
|
for item in source_path.iterdir():
|
|
if item.name.endswith(".item"):
|
|
item.unlink()
|
|
(source_path / "state").unlink(missing_ok=True)
|
|
|
|
|
|
@pytest.fixture
|
|
def using_source() -> Generator[Callable, None, LocalSource]:
|
|
test_data = Path(__file__).parent
|
|
sources: list[Path] = []
|
|
|
|
def _using_source(name: str):
|
|
source_path = test_data / name
|
|
clean_source(source_path)
|
|
sources.append(source_path)
|
|
return LocalSource(test_data, name)
|
|
|
|
yield _using_source
|
|
|
|
for source_path in sources:
|
|
clean_source(source_path)
|