Фильтрация набора для соответствия перестановкам строк ⇐ Python

[Anonymous]

Я пытаюсь использовать itertools.permutations(), чтобы вернуть все перестановки строки и вернуть только те, которые являются членами набора слова.

Код: Выделить всё

import itertools

def permutations_in_dict(string, words):
'''
Parameters
----------
string : {str}
words : {set}

Returns
-------
list : {list} of {str}

Example
-------
>>> permutations_in_dict('act', {'cat', 'rat', 'dog', 'act'})
['act', 'cat']
'''

Мое текущее решение отлично работает в терминале, но почему-то не смог пройти тестовый пример...

Код: Выделить всё

return list(set([''.join(p) for p in itertools.permutations(string)]) & words)

Любая помощь будет оценена по достоинству.

Подробнее здесь: https://stackoverflow.com/questions/448 ... rmutations