class testFizzBuzz(unittest.TestCase): def setUp(self): self.fizzbuzz = FizzBuzz() def test_one_should_return_one(self): self.fizzbuzz.set_num(1) expected = '1' actual = self.fizzbuzz.get() self.assertEqual(expected, actual) def test_two_should_return_two(self): self.fizzbuzz.set_num(2) expected = '2' actual = self.fizzbuzz.get() self.assertEqual(expected, actual) def test_three_should_return_fizz(self): self.fizzbuzz.set_num(3) expected = 'fizz' actual = self.fizzbuzz.get() self.assertEqual(expected, actual) def test_five_should_return_buzz(self): self.fizzbuzz.set_num(5) expected = 'buzz' actual = self.fizzbuzz.get() self.assertEqual(expected, actual) def test_fifteen_should_return_fizzbuzz(self): self.fizzbuzz.set_num(15) expected = 'fizzbuzz' actual = self.fizzbuzz.get() self.assertEqual(expected, actual)