'''
   primecheckertester.py
   Jeff Ondich, 9 May 2012

   A sample unit test using the PyUnit / unittest module.
'''

import primechecker
import unittest

class PrimeCheckerTester(unittest.TestCase):
    def setUp(self):
        self.primeChecker = primechecker.PrimeChecker(100)

    def testNegative(self):
        self.assertFalse(self.primeChecker.isPrime(-4))

    def testZero(self):
        self.assertFalse(self.primeChecker.isPrime(0))

    def testOne(self):
        self.assertFalse(self.primeChecker.isPrime(1))

    def testTwo(self):
        self.assertTrue(self.primeChecker.isPrime(2))

    def testPrime(self):
        self.assertTrue(self.primeChecker.isPrime(97))

    def testComposite(self):
        self.assertFalse(self.primeChecker.isPrime(96))

    def testAboveRange(self):
        self.assertTrue(self.primeChecker.isPrime(107))

    def testPrimesBelow(self):
        self.assertEqual(self.primeChecker.getPrimesBelow(20), [2, 3, 5, 7, 11, 13, 17, 19])

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