def test_get_words_error():
"""Ensure that we throw an exception on encountering an error while trying
to read the words from the dictionary file.
"""
with pytest.raises(SystemExit):
args = {
"-d": "doesn't exist.txt",
}
_get_words(args)
def test_get_words():
"""Ensure that we read the word list from a file."""
word_file_contents = """
foo
bar
"""
with tempfile.NamedTemporaryFile("w+") as f:
f.write(word_file_contents)
f.seek(0)
args = {
"-d": f.name,
}
assert _get_words(args) == {"foo", "bar"}
