def test_group_keys_with_some_wrong_by_provider(self): keys = ['movie:tt1234567890:imdb', 'movie:tt1234567891:imdb', 'movie:tt1234567892_imdb', 'movie:1234567893:imdb'] groups = Movie.group_keys_by_provider(keys=keys) self.assertEqual(len(groups), 1) self.assertEqual(len(groups[0]['keys']), 2) self.assertEqual(groups[0]['parser'], None) keys = ['movie:tt1234567890:imdb', 'movie:tt1234567891:imdb', 'movie:tt1234567892_imdb', 'movie:1234567893:imdb'] groups = Movie.group_keys_by_provider(keys=keys, make_parser=True) self.assertEqual(len(groups), 1) self.assertEqual(len(groups[0]['keys']), 2) self.assertTrue(isinstance(groups[0]['parser'], IMDBParser))
def test_group_empty_keys_list_by_provider(self): keys = [] groups = Movie.group_keys_by_provider(keys=keys) self.assertEqual(len(groups), 0)