Coverage for lasso/io/test_files.py: 100%

1import unittest 

2from lasso.io.files import collect_files 

3 

4 

5class Test(unittest.TestCase): 

6 def test_collect_files(self): 

7 files = collect_files("test/io_test", "*.txt") 

8 self.assertEqual(len(files), 1) 

9 

10 files = collect_files("test/io_test/", "*.txt", recursive=True) 

11 self.assertEqual(len(files), 2) 

12 

13 files1, files2 = collect_files("test/io_test/", ["*.txt", "*.yay"], recursive=True) 

14 self.assertEqual(len(files1), 2) 

15 self.assertEqual(len(files2), 1) 

16 

17 files1, files2 = collect_files( 

18 ["test/io_test/", "test/io_test/subfolder"], ["*.txt", "*.yay"] 

19 ) 

20 self.assertEqual(len(files1), 2) 

21 self.assertEqual(len(files2), 1)