Welcome to the Treehouse Community

Want to collaborate on code errors? Have bugs you need feedback on? Looking for an extra set of eyes on your latest project? Get support with fellow developers, designers, and programmers of all backgrounds and skill levels here with the Treehouse Community! While you're at it, check out some resources Treehouse students have shared here.

Looking to learn something new?

Treehouse offers a seven day free trial for new students. Get access to thousands of hours of content and join thousands of Treehouse students and alumni in the community today.

Start your free trial

Python Python Testing Be Assertive assertIn

Devin Scheu
Devin Scheu
66,191 Points

Python Tests Help

Question:The get_anagrams() function takes one or more words and returns anagrams for each of them as a list. Finish the test_in_anagrams() test to check that the anagrams for the string "treehouse" contains the word "house".

I'm pretty sure it's just that I'm not understanding this fully. I'm going to go back through the course after this post and see if I can get a better understanding of this testing course. I'm fairly new to python testing so anything helps :).

Thanks, -Devin

Code:

tests.py
import unittest

from string_fun import get_anagrams


class AnagramTests(unittest.TestCase):
    def test_in_anagrams(self):
          get_anagrams(assertIn("house", "treehouse"))
string_fun.py
import itertools


def is_palindrome(yarn):
    """Return whether or not a string is a palindrome.

    A palindrome is a word/phrase that's the same in
    both directions.
    """
    return yarn == yarn[::-1]


def get_anagrams(*yarn):
    """Return a list of anagrams for a string."""
    # If only one letter came in, return it
    if yarn:
        if len(yarn[0]) <= 1:
            return list(yarn)
    else:
        raise ValueError("Must provide at least two letters")

    # Get all of the words from the dictionary
    words = set([
        w.strip().lower() for w in open('words.txt')
    ])

    output = set()
    for thread in yarn:
        thread = thread.lower()
        # Find all possible anagrams
        for i in range(2, len(thread)):
            fibers = set(
                [''.join(w) for w in itertools.permutations(thread, i)]
            )
            output.update(fibers.intersection(words))

    # Finally, return all of the combinations that are in the dictionary
    return sorted(list(output))

6 Answers

Dan Johnson
Dan Johnson
40,532 Points

assertIn looks to see if a specific item is in a collection. So you'll want assertIn to be the outermost call (called from self). Supply this call with "house" for the item you're looking for, and the collection of all anagrams from get_anagrams.

Devin Scheu
Devin Scheu
66,191 Points

Here's my updated code, I'm still getting it wrong but I think you led me onto the right track.

import unittest

from string_fun import get_anagrams


class AnagramTests(unittest.TestCase):
    def test_in_anagrams(self):
        self.assertIn("house", get_anagrams())
Devin Scheu
Devin Scheu
66,191 Points

I figured it out, all I had to do is pass Treehouse as a string to get_anagrams, thanks for the help!

Andrew Winkler
Andrew Winkler
37,739 Points

This worked for me:

import unittest

from string_fun import get_anagrams


class AnagramTests(unittest.TestCase):
    def test_in_anagrams(self):
      self.assertIn('house', get_anagrams('treehouse'))

task2 class AnagramTestCase(unittest.TestCase): def test_not_in_anagrams(self): with self.assertNotIn(('code', get_anagrams('Treehouse')): get_anagrams('Treehouse')

import unittest

from string_fun import get_anagrams

class AnagramTests(unittest.TestCase): def test_in_anagrams(self): self.assertIn("house", get_anagrams('Treehouse'))

import unittest

from string_fun import get_anagrams

class AnagramTests(unittest.TestCase): def test_in_anagrams(self): self.assertIn("house", get_anagrams('Treehouse'))

Neil Gordon
Neil Gordon
8,823 Points
import unittest

from string_fun import get_anagrams


class AnagramTests(unittest.TestCase):
    def test_in_anagrams(self):
        self.assertIn("house", get_anagrams('Treehouse'))
        pass