def test_get_filenames_ignore_regex(self):
self.assertTrue(
len(
list(
fileio.get_filenames(self.tests_dir,
ignore_regex='test_',
ignore_invisible=True))) == 0)
def test_get_filenames_match_regex(self):
self.assertTrue(
len(
list(
fileio.get_filenames(self.tests_dir,
match_regex='fileio',
extension='.py'))) == 1)
def test_get_filenames(self):
expected = sorted(os.path.join(self.tests_dir, fname)
for fname in os.listdir(self.tests_dir)
if os.path.isfile(os.path.join(self.tests_dir, fname)))
observed = sorted(fileio.get_filenames(self.tests_dir,
ignore_invisible=False,
recursive=False))
self.assertEqual(observed, expected)
def test_get_filenames(self):
expected = sorted(os.path.join(self.tests_dir, fname)
for fname in os.listdir(self.tests_dir)
if os.path.isfile(os.path.join(self.tests_dir, fname)))
observed = sorted(fileio.get_filenames(self.tests_dir,
ignore_invisible=False,
recursive=False))
self.assertEqual(observed, expected)
def filenames(self):
"""
Tuple[str]: Full paths on disk for all Reddit comments files found under
the ``data_dir`` directory, sorted chronologically.
"""
if os.path.exists(self.data_dir):
return tuple(sorted(fileio.get_filenames(self.data_dir, extension='.bz2', recursive=True)))
else:
LOGGER.warning(
'%s data directory does not exist', self.data_dir)
return tuple()
def test_get_filenames_match_regex(self):
self.assertTrue(
len(list(fileio.get_filenames(self.tests_dir,
match_regex='fileio',
extension='.py'))) == 1)
def test_get_filenames_ignore_regex(self):
self.assertTrue(
len(list(fileio.get_filenames(self.tests_dir,
ignore_regex='test_',
ignore_invisible=True))) == 0)
def test_get_filenames_ignore_invisible(self):
path = os.path.dirname(os.path.abspath(__file__))
self.assertTrue(
len(list(fileio.get_filenames(path, ignore_invisible=True))) <=
len(list(fileio.get_filenames(path, ignore_invisible=False)))
)
def test_get_filenames_ignore_invisible(self):
path = os.path.dirname(os.path.abspath(__file__))
self.assertTrue(
len(list(fileio.get_filenames(path, ignore_invisible=True))) <=
len(list(fileio.get_filenames(path, ignore_invisible=False))))
def test_get_filenames_ignore_invisible(self):
self.assertTrue(
len(list(fileio.get_filenames(self.tests_dir, ignore_invisible=True))) \
< len(list(fileio.get_filenames(self.tests_dir, ignore_invisible=False))))
def test_get_filenames_ignore_invisible(self):
self.assertTrue(
len(list(fileio.get_filenames(self.tests_dir, ignore_invisible=True)))
< len(list(fileio.get_filenames(self.tests_dir, ignore_invisible=False)))
)
