Test suite for anagrams.py, at https://gist.github.com/thebenedict/7655053

test_anagrams.py

import unittest
from StringIO import StringIO
from anagrams import find_anagrams

class AnagramsTestCase(unittest.TestCase):
  """Tests for anagrams.py"""

  def test_finds_simple_anagrams(self):
    """Does it find anagrams for 'art'?"""
    anas = find_anagrams(StringIO("art\nrat\ntar"))
    self.assertEqual({"art": ["art", "rat", "tar"]}, anas)

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