test_numbthy_unittest.py

######################################################################################
# Test NUMBTHY.PY
# Basic Number Theory functions implemented in Python
# Author: Robert Campbell, <campbell@math.umbc.edu>
# Date: 14 Oct 2014
# Version 0.8
######################################################################################

import unittest
import numbthy
import random

class Test_numbthy(unittest.TestCase):

    def test_gcd(self):
        for testcase in ((1,1,1),(-1,1,1),(6,0,6),(0,6,6),(6,6,6),(1234,5678,2),(12345675,34567890,2469135)):
            self.assertEqual(numbthy.gcd(testcase[0],testcase[1]),testcase[2])

    def test_xgcd(self):
        for testcase in ((1,-1,1,0,-1),(6,8,2,-1,1),(12345,2345,5,87,-458),(98760,76540,20,1898,-2449)):
            self.assertEqual(numbthy.xgcd(testcase[0],testcase[1]),(testcase[2],testcase[3],testcase[4]))

    def test_xgcd_random(self):
        for testnum in range(10):
            a = random.randint(0,10**20); b = random.randint(0,10**20); 
            (g,x,y) = numbthy.xgcd(a,b)
            try:
                self.assertEqual(g,a*x+b*y)
                break
            except AssertionError:
                raise AssertionError("***** Error in xgcd *****: {2} != ({0})*({3}) + ({1})*({4})".format(a,b,g,x,y))

    def test_power_mod(self):
        for testcase in ((2,5,13,6),(2,-21,31,16),(-2,-21,31,15),(0,5,31,0),(5,0,31,1)):
            self.assertEqual(numbthy.power_mod(testcase[0],testcase[1],testcase[2]),testcase[3])

    def test_power_mod_raise_ValueError(self):  # Error for power_mod(2,-11,26)
        self.assertRaises(ValueError,numbthy.power_mod,2,-11,26)

    def test_power_mod_raise_ZeroDivisionError(self):  # Error for power_mod(2,5,0)
        self.assertRaises(ZeroDivisionError,numbthy.power_mod,2,5,0)

    def test_is_prime(self):
        for testcase in ((1,False),(0,False),(-2,True),(19,True),(1236,False),(1237,True),(321197185,False)):
            self.assertEqual(numbthy.is_prime(testcase[0]),testcase[1])

    def test_euler_phi(self):
        for testcase in ((-5,0),(0,0),(1,1),(2,1),(4,2),(8*7,24),(2**4*3**3*7**2*19,108864)):
            self.assertEqual(numbthy.euler_phi(testcase[0]),testcase[1])

    def test_carmichael_lambda(self):
        for testcase in ((1,1),(2,1),(4,2),(8*7,6),(2**4*3**3*7**2*19,252)):
            self.assertEqual(numbthy.carmichael_lambda(testcase[0]),testcase[1])

    def test_carmichael_lambda_raise_ValueError(self):  # Error for carmichael_lambda(0)
        self.assertRaises(ValueError,numbthy.carmichael_lambda,0)

    def test_factor(self):
        for testcase in ((-15,((3, 1), (5, 1))),(1234561000,((2, 3), (5, 3), (211, 1), (5851, 1)))):
            self.assertEqual(numbthy.factor(testcase[0]),testcase[1])

if __name__ == '__main__':
    unittest.main()