rmotr.com assignment 3

find_anagrams.py

import unittest


def find_anagrams(list_of_strings, word):
    anagrams = []
    for test_word in list_of_strings:
        if ''.join(sorted(test_word)) == ''.join(sorted(word)):
            anagrams.append(test_word)
    return anagrams


class AnagramsTestCase(unittest.TestCase):
    def test_multiple_matches(self):
        self.assertEqual(find_anagrams(["spare", "hello", "pears", "world", "reaps"], "parse"),
                         ["spare", "pears", "reaps"])

    def test_one_match(self):
        self.assertEqual(find_anagrams(["spared", "hello", "appears", "world", "reaps"], "parse"),
                         ["reaps"])

    def test_no_matches(self):
        self.assertEqual(find_anagrams(["spared", "hello", "appears", "world", "teaps"], "parse"),
                         [])