mirror of
https://github.com/donnemartin/interactive-coding-challenges.git
synced 2024-03-22 13:11:13 +08:00
25 lines
675 B
Python
25 lines
675 B
Python
import unittest
|
|
|
|
|
|
class TestSolution(unittest.TestCase):
|
|
|
|
def test_can_win_nim(self):
|
|
solution = Solution()
|
|
self.assertRaises(TypeError, solution.can_win_nim, None)
|
|
self.assertEqual(solution.can_win_nim(1), True)
|
|
self.assertEqual(solution.can_win_nim(2), True)
|
|
self.assertEqual(solution.can_win_nim(3), True)
|
|
self.assertEqual(solution.can_win_nim(4), False)
|
|
self.assertEqual(solution.can_win_nim(7), True)
|
|
self.assertEqual(solution.can_win_nim(40), False)
|
|
print('Success: test_can_win_nim')
|
|
|
|
|
|
def main():
|
|
test = TestSolution()
|
|
test.test_can_win_nim()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|