def test_torrents_missing_episode_or_tvdb_imdb_id(self): torrents = [{}] result = ti.deduplicate_torrents(torrents) self.assertEqual(result, []) torrents[0]['episode_info'] = {} result = ti.deduplicate_torrents(torrents) self.assertEqual(result, []) torrents[0]['episode_info'] = {'tvdb': '12345'} self.assertEqual(result, [])
def test_torrents_are_not_episodes(self): torrents = [{ 'title': 'FooBar', 'episode_info': { 'tvdb': '12345', 'imdb': 'tt12345' } }] result = ti.deduplicate_torrents(torrents) self.assertEqual(result, [])
def test_torrents_are_episodes(self): torrents1 = [{ 'title': 'FooBar', 'episode_info': { 'tvdb': '12345', 'imdb': 'tt12345', 'seasonnum': '01', 'epnum': '01' } }] result = ti.deduplicate_torrents(torrents1) self.assertEqual(result, torrents1) torrents2 = [{ 'title': 'Foo.s01e01.mp4', 'episode_info': { 'tvdb': '12345', 'imdb': 'tt12345' } }] result = ti.deduplicate_torrents(torrents2) self.assertEqual(result[0]['episode_info']['seasonnum'], '01') self.assertEqual(result[0]['episode_info']['epnum'], '01')
def test_differentiating_sd_and_hd_torrents(self): torrents = [ { 'title': 'Foo.s01e01.mp4', 'episode_info': { 'tvdb': '12345', 'imdb': 'tt12345' }, 'seeders': 10 }, { 'title': 'Foo.s01e01.720p.mkv', 'episode_info': { 'tvdb': '12345', 'imdb': 'tt12345' }, 'seeders': 20 }, ] result = ti.deduplicate_torrents(torrents) self.assertEqual(len(result), 2)
def test_deduplication_based_on_repack(self): torrents = [ { 'title': 'Foo.s01e01.mp4', 'episode_info': { 'tvdb': '12345', 'imdb': 'tt12345' }, 'seeders': 20 }, { 'title': 'Foo.s01e01.REPACK.mp4', 'episode_info': { 'tvdb': '12345', 'imdb': 'tt12345' }, 'seeders': 10 }, ] result = ti.deduplicate_torrents(torrents) self.assertEqual(len(result), 1) self.assertEqual(result[0]['seeders'], 10)