def test_walktree_real(self):
tmpdir = tempfile.mkdtemp(suffix='classify.test')
path = os.path.join(tmpdir, self.REAL_EPISODE)
name = ' '.join(self.REAL_EPISODE.split('.')[:3])
fd = open(path, "w+")
fd.close()
analyzer = SeriesAnalyzer(tmpdir)
analyzer.extract()
self.assertEquals(1, len(analyzer.get_infos()),
"Incorrectly matched 1 episode!")
found_name = analyzer.get_infos()[0]['name']
self.assertEquals(name, found_name,
"Incorrectly matched episode name: %s!" % found_name)
def create_tree(self):
self.dir = tempfile.mkdtemp(suffix='-test')
for i in range(0, self.FILE_NUM):
name = randstr()
season = random.randint(1, 10)
episode = random.randint(1, 24)
real_name = self.FORMAT % (name, season, episode)
path = os.path.join(self.dir, real_name)
self.real_files.append(real_name)
self.real_infos[name] = {
'path': path,
'name': name,
'season': int(season),
'episode': int(episode)
}
fd = open(path, mode='w+')
fd.close()
self.analyzer = SeriesAnalyzer(self.dir)
self.analyzer.extract()
