Написание doctests для метода, который сокращает словарь путем поиска переданного ключевого слова в ключах исходного словаря и возврата нового сокращенного словаря. Моя строка документации выглядит следующим образом:
def abbreviate_dict(key_word, original_dict):
"""
>>> orig_dict = {apple_stems: 2, apple_cores: 5, apple_seeds: 3}
>>> abbreviate_dict('apple', orig_dict)
{'cores': 5, 'seeds': 3, 'stems': 2}
"""
etc.
return new_dict
Функция работает, но когда я запускаю doctest py.test, функция не проходит тест, поскольку возвращает строки в формате unicode. Я программно не переключаю строки на юникод в своей функции, но я знаю, что python 2.7 возвращает в юникоде.
Expected:
{'cores': 5, 'seeds': 3, 'stems': 2}
Got:
{u'cores': 5, u'seeds': 3, u'stems': 2}
Как я могу заставить doctest признать, что Unicode и обычные строковые выходные данные одинаковы?