Coverage for src/hallmd/data/spt100/__init__.py: 96%

1from importlib import resources 

2from pathlib import Path 

3 

4from . import diamant2014 as _diamant2014 

5from . import macdonald2019 as _macdonald2019 

6from . import sankovic1993 as _sankovic1993 

7 

8 

9def diamant2014(datasets: list[str] | str | None = None) -> list[Path]: 

10 dir = resources.files(_diamant2014) 

11 

12 if datasets is None: 

13 datasets = ["L3", "aerospace"] 

14 elif isinstance(datasets, str): 

15 datasets = [datasets] 

16 

17 data_paths: list[Path] = [] 

18 for set in datasets: 

19 with resources.as_file(dir / f"data_{set}.csv") as path: 

20 data_paths.append(path) 

21 

22 return data_paths 

23 

24 

25def macdonald2019() -> list[Path]: 

26 dir = resources.files(_macdonald2019) 

27 with resources.as_file(dir / "data.csv") as path: 

28 return [path] 

29 

30 

31def sankovic1993() -> list[Path]: 

32 dir = resources.files(_sankovic1993) 

33 with resources.as_file(dir / "data.csv") as path: 

34 return [path] 

35 

36 

37def all() -> list[Path]: 

38 return sankovic1993() + macdonald2019() + diamant2014()