class FizzBuzzTest(unittest.TestCase):
def setUp(self):
self.app = FizzBuzz()
def test_returns_one_when_one(self):
self.assertEqual(self._submit_value(1), 1, "1 should return 1")
def test_returns_fizz_when_three(self):
self.assertEqual(self._submit_value(3), 'fizz')
def test_returns_fizz_when_six(self):
self.assertEqual(self._submit_value(6), 'fizz')
def test_returns_buzz_when_five(self):
self.assertEqual(self._submit_value(5), 'buzz')
def test_returns_buzz_when_ten(self):
self.assertEqual(self._submit_value(10), 'buzz')
def test_returns_four_when_four(self):
self.assertEqual(self._submit_value(4), 4)
def test_returns_fizzbuzz_when_15(self):
self.assertEqual(self._submit_value(15), 'fizzbuzz')
def _submit_value(self, value):
return self.app.do_it(value=value)
