def test_works_with_str_basedir(self): r5 = self.tempdir / 's05' r5.mkdir(parents=True) r1 = self.tempdir / 's01-toast' r1.mkdir(parents=True) r2 = self.tempdir / 's02-grr' r2.mkdir(parents=True) res = list(utils.find_tiledirs(str(self.tempdir))) self.assertEqual(res, [(1, r1), (2, r2), (5, r5)])
def test_condition_filter_case_insensitive(self): r5 = self.tempdir / 's05-H7-1' r5.mkdir(parents=True) r1 = self.tempdir / 's01-wTb-2' r1.mkdir(parents=True) r2 = self.tempdir / 's02-wtc-2' r2.mkdir(parents=True) res = list(utils.find_tiledirs(self.tempdir, conditions=['WTB', 'h7'])) self.assertEqual(res, [(1, r1), (5, r5)])
def test_can_filter_on_multiple_conditions(self): r5 = self.tempdir / 's05-h7-1' r5.mkdir(parents=True) r1 = self.tempdir / 's01-wtb-2' r1.mkdir(parents=True) r2 = self.tempdir / 's02-wtc-2' r2.mkdir(parents=True) res = list(utils.find_tiledirs(self.tempdir, conditions=['wtb', 'h7'])) self.assertEqual(res, [(1, r1), (5, r5)])
def test_can_filter_on_single_tile_number_string(self): r5 = self.tempdir / 's05' r5.mkdir(parents=True) r1 = self.tempdir / 's01-toast' r1.mkdir(parents=True) r2 = self.tempdir / 's02-grr' r2.mkdir(parents=True) res = list(utils.find_tiledirs(self.tempdir, tiles='2')) self.assertEqual(res, [(2, r2)])
def test_ignores_unparsable_dirs(self): r5 = self.tempdir / 's05' r5.mkdir(parents=True) r1 = self.tempdir / 's01-toast' r1.mkdir(parents=True) r2 = self.tempdir / 's02-grr' r2.mkdir(parents=True) bad = self.tempdir / 'agkjslakdhjfld' bad.mkdir(parents=True) res = list(utils.find_tiledirs(self.tempdir)) self.assertEqual(res, [(1, r1), (2, r2), (5, r5)])
def test_ignores_files(self): r5 = self.tempdir / 's05' r5.mkdir(parents=True) r1 = self.tempdir / 's01-toast' r1.mkdir(parents=True) r2 = self.tempdir / 's02-grr' r2.mkdir(parents=True) bad = self.tempdir / 's04-bad' with bad.open('wt') as fp: fp.write('bad') res = list(utils.find_tiledirs(self.tempdir)) self.assertEqual(res, [(1, r1), (2, r2), (5, r5)])
def test_finds_one_tile_with_condition(self): r1 = self.tempdir / 's03-ponies' r1.mkdir(parents=True) res = list(utils.find_tiledirs(self.tempdir)) self.assertEqual(res, [(3, r1)])
def test_finds_no_tiles(self): res = list(utils.find_tiledirs(self.tempdir)) self.assertEqual(res, [])