def test_make_dataset_multiple_extensions(self, mock_walk):
mock_walk.return_value = [
('', (), ('b.1', 'c.2'))
]
out = ds.make_dataset('a',['1','2'])
self.assertTrue(out == ['b.1', 'c.2'])
def test_make_dataset_all_files(self, mock_walk):
mock_walk.return_value = [
('/root', (), ('b.a', 'c.a'))
]
out = ds.make_dataset('a',['a'])
self.assertTrue(out == ['/root/b.a', '/root/c.a'])
def test_make_dataset_no_root(self, mock_walk):
mock_walk.return_value = [
('', (), ('b.a', 'c.a'))
]
out = ds.make_dataset('a',['a'])
self.assertTrue(out == ['b.a', 'c.a'])
def test_make_dataset_no_files(self, mock_walk):
mock_walk.return_value = [('/root', (), ('b.v', 'c.v'))]
out = ds.make_dataset('a', ('a'))
self.assertTrue(out == [])