Beispiel #1
0
    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))
Beispiel #2
0
    def test_group_empty_keys_list_by_provider(self):
        keys = []

        groups = Movie.group_keys_by_provider(keys=keys)
        self.assertEqual(len(groups), 0)