''' Created on Aug 16, 2018 @author: haidong ''' import unittest from sources.fiblastdigit import getFibLastDigitNaive class Test(unittest.TestCase): def testName(self): self.assertEqual(0, getFibLastDigitNaive(0)) self.assertEqual(1, getFibLastDigitNaive(1)) self.assertEqual(2, getFibLastDigitNaive(3)) self.assertEqual(5, getFibLastDigitNaive(40)) self.assertEqual(9, getFibLastDigitNaive(331)) self.assertEqual(5, getFibLastDigitNaive(327305)) if __name__ == "__main__": #import sys;sys.argv = ['', 'Test.testName'] unittest.main()