FizzBuzzTest_AgileMaine.py

from unittest import TestCase


def fizzbuzz(i):
    #FIZZ when number 3
    if i % 3 == 0 and i % 5 == 0:
        return "FIZZBUZZ"
    if i % 3 == 0:
        return "FIZZ"
    if i % 5 == 0:
        return "BUZZ"
    return str(i)


class FizzBuzzTest(TestCase):
    def test_1(self):
        #when we ask for 1
        result = fizzbuzz(1)
        #verify result = 1
        self.assertEquals(result,"1")

    def test_2(self):
        result = fizzbuzz(2)
        self.assertEquals(result, "2")

    def test_3(self):
        result = fizzbuzz(3)
        self.assertEquals(result, "FIZZ")

    def test_5(self):
        result = fizzbuzz(5)
        self.assertEquals(result, "BUZZ")

    def test_6(self):
        result = fizzbuzz(6)
        self.assertEquals(result, "FIZZ")

    def test_10(self):
        result = fizzbuzz(10)
        self.assertEquals(result, "BUZZ")

    def test_15(self):
        result = fizzbuzz(15)
        self.assertEquals(result, "FIZZBUZZ")