import unittest
import lgprime 

class TestLargePrime(unittest.TestCase):
    def setUp(self):
        self.lgPrime = lgprime.lgPrime()

    def testInitial(self):
        self.lgPrime.setNumber(13195)
        assert self.lgPrime.lgPrimeNum() == (29), "incorrect answer %d" % lgPrime.lgPrimeNum()

    def testFinal(self):
        self.lgPrime.setNumber(4000000)
        assert self.lgPrime.lgPrimeNum() == (4613732), "incorrect answer %d" % self.lgPrime.lgPrimeNum()

lgPrimeTestSuite = unittest.TestSuite()

lgPrimeTestSuite.addTest(TestLargePrime("testInitial"))
lgPrimeTestSuite.addTest(TestLargePrime("testFinal"))

runner = unittest.TextTestRunner()
runner.run(lgPrimeTestSuite)

import unittest
import lgprime 

class TestLargePrime(unittest.TestCase):
    def setUp(self):
        self.lgPrime = lgprime.lgPrime()

    def testInitial(self):
        self.lgPrime.setNumber(13195)
        assert self.lgPrime.lgPrimeNum() == (29), "incorrect answer %d" % lgPrime.lgPrimeNum()

    def testFinal(self):
        self.lgPrime.setNumber(600851475143)
        assert self.lgPrime.lgPrimeNum() == (0), "incorrect answer %d" % self.lgPrime.lgPrimeNum()

lgPrimeTestSuite = unittest.TestSuite()

lgPrimeTestSuite.addTest(TestLargePrime("testInitial"))
lgPrimeTestSuite.addTest(TestLargePrime("testFinal"))

runner = unittest.TextTestRunner()
runner.run(lgPrimeTestSuite)

update number for final Project Euler answer