class FizzBuzzTest(unittest.TestCase): def setUp(self): self.fizzbuzz = FizzBuzz() def test_input_number_should_return_same_number(self): result = self.fizzbuzz.take(1) self.assertEqual(result, 1) result = self.fizzbuzz.take(2) self.assertEqual(result, 2) def test_input_number_divisible_by_3_should_return_fizz(self): result = self.fizzbuzz.take(3) self.assertEqual(result, 'fizz') result = self.fizzbuzz.take(6) self.assertEqual(result, 'fizz') def test_input_number_divisible_by_5_should_return_buzz(self): result = self.fizzbuzz.take(5) self.assertEqual(result, 'buzz') result = self.fizzbuzz.take(10) self.assertEqual(result, 'buzz') def test_input_number_divisible_by_3_and_5_should_return_fizzbuzz(self): result = self.fizzbuzz.take(15) self.assertEqual(result, 'fizzbuzz') result = self.fizzbuzz.take(30) self.assertEqual(result, 'fizzbuzz') result = self.fizzbuzz.take(0) self.assertEqual(result, 'fizzbuzz')