mirror of
https://github.com/donnemartin/interactive-coding-challenges.git
synced 2024-03-22 13:11:13 +08:00
37 lines
772 B
Python
37 lines
772 B
Python
from nose.tools import assert_equal, assert_raises
|
|
|
|
|
|
class TestFizzBuzz(object):
|
|
|
|
def test_fizz_buzz(self):
|
|
solution = Solution()
|
|
assert_raises(TypeError, solution.fizz_buzz, None)
|
|
assert_raises(ValueError, solution.fizz_buzz, 0)
|
|
expected = [
|
|
'1',
|
|
'2',
|
|
'Fizz',
|
|
'4',
|
|
'Buzz',
|
|
'Fizz',
|
|
'7',
|
|
'8',
|
|
'Fizz',
|
|
'Buzz',
|
|
'11',
|
|
'Fizz',
|
|
'13',
|
|
'14',
|
|
'FizzBuzz'
|
|
]
|
|
assert_equal(solution.fizz_buzz(15), expected)
|
|
print('Success: test_fizz_buzz')
|
|
|
|
|
|
def main():
|
|
test = TestFizzBuzz()
|
|
test.test_fizz_buzz()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main() |