Test case for generator.py

test_generator.py

import unittest

from buzz import generator

def test_sample_single_word():
    l = ('foo', 'bar', 'foobar')
    word = generator.sample(l)
    assert word in l

def test_sample_multiple_words():
    l = ('foo', 'bar', 'foobar')
    words = generator.sample(l, 2)
    assert len(words) == 2
    assert words[0] in l
    assert words[1] in l
    assert words[0] is not words[1]

def test_generate_buzz_of_at_least_five_words():
    phrase = generator.generate_buzz()
    assert len(phrase.split()) >= 5

ImportError while importing test module /cicd-buzz/tests/test_generator.py'.
Hint: make sure your test modules/packages have valid Python names.
Traceback:
test_generator.py:3: in
from buzz import generator
E ImportError: No module named buzz