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)
class TestFizzBuzz(unittest.TestCase): def setUp(self): self.fizzbuzz = FizzBuzz() def test_number_three_is_fizz(self): self.assertEquals(self.fizzbuzz.get(1), 1, 'Number 1 is 1') self.assertEquals(self.fizzbuzz.get(3), 'fizz', 'Number 3 is "fizz"') self.assertEquals(self.fizzbuzz.get(33), 'fizz', 'Number 33 is "fizz"') def test_number_five_is_buzz(self): self.assertEquals(self.fizzbuzz.get(5), 'buzz', 'Number 5 is "buzz"') self.assertEquals(self.fizzbuzz.get(20), 'buzz', 'Number 20 is "buzz"') def test_number_three_and_five_is_fizzbuzz(self): self.assertEquals(self.fizzbuzz.get(15), 'fizzbuzz', 'Number 15 is "fizzbuzz"') self.assertEquals(self.fizzbuzz.get(30), 'fizzbuzz', 'Number 20 is "fizzbuzz"')