"""
------------------------------------------------------------------------------
--                                                                          --
--                        MARCH 5 LECTURE EXAMPLES                          --
--                                                                          --
--                      M A R C H 5 _ T E S T S . P Y                       --
--                                                                          --
------------------------------------------------------------------------------
-- Jeremy D Keffer                                                          --
--                                                                          --
-- CISC106 011 Spring 2013                                                  --
--                                                                          --
------------------------------------------------------------------------------
-- This file contains unit tests for the functions we wrote in class on     --
-- March 5.                                                                 --
--                                                                          --
------------------------------------------------------------------------------
"""

from march5 import *
import unittest
import math

class TestMarch5(unittest.TestCase):
    def test_is_prime(self):
        self.assertFalse(is_prime(1))
        self.assertTrue(is_prime(2))
        self.assertTrue(is_prime(3))
        self.assertFalse(is_prime(4))
        self.assertTrue(is_prime(5))

    def test_factorial(self):
        self.assertEqual(factorial(5), 120)
        self.assertEqual(factorial(6), 720)
        self.assertEqual(factorial(4), 24)
        
if __name__ == '__main__':
    try:
        unittest.main()
    except SystemExit:
        pass