Quiz de listas do pug-pe challenge

gistfile1.py

import unittest

def search_list(data, target):
    return [[i for i, d in enumerate(data) if d == t] for t in target]

class TestIt(unittest.TestCase):

    def testIt(self):
        self.assertEqual([[0, 2], [10], [5, 6, 7]],
                        search_list([1,4,1,4, 6, 5, 5, 5, 4, 2, 3], [1, 3, 5]))

if __name__ == '__main__':
    unittest.main()