mirror of
https://github.com/donnemartin/interactive-coding-challenges.git
synced 2024-03-22 13:11:13 +08:00
23 lines
536 B
Python
23 lines
536 B
Python
import unittest
|
|
|
|
|
|
class TestMath(unittest.TestCase):
|
|
|
|
def test_check_prime(self):
|
|
math = Math()
|
|
self.assertRaises(TypeError, math.check_prime, None)
|
|
self.assertRaises(TypeError, math.check_prime, 98.6)
|
|
self.assertEqual(math.check_prime(0), False)
|
|
self.assertEqual(math.check_prime(1), False)
|
|
self.assertEqual(math.check_prime(97), True)
|
|
print('Success: test_check_prime')
|
|
|
|
|
|
def main():
|
|
test = TestMath()
|
|
test.test_check_prime()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|