Coverage for src/rechunk_data/tests/test_search_files.py: 100%

1"""Unit tests for searching for files.""" 

2from pathlib import Path 

3 

4from rechunk_data._rechunk import _search_for_nc_files 

5 

6 

7def test_search_directory(data_dir: Path) -> None: 

8 """Search a directory with valid netcdf files.""" 

9 data_files = sorted(_search_for_nc_files(data_dir)) 

10 assert len(data_files) == 9 

11 search_dir_with_wild_cards = data_dir / "*" 

12 data_files = sorted(_search_for_nc_files(search_dir_with_wild_cards)) 

13 assert len(data_files) == 9 

14 search_dir_with_wild_cards = data_dir / "*.txt" 

15 data_files = sorted(_search_for_nc_files(search_dir_with_wild_cards)) 

16 assert len(data_files) == 0 

17 

18 

19def test_search_directory_with_wrong_files_type(wrong_file_type: Path) -> None: 

20 """Search a directory contaning files not matching the suffixes.""" 

21 data_files = sorted(_search_for_nc_files(wrong_file_type)) 

22 assert len(data_files) == 0 

23 search_dir_with_wild_cards = wrong_file_type / "*" 

24 data_files = sorted(_search_for_nc_files(search_dir_with_wild_cards)) 

25 assert len(data_files) == 0