''' Created on Aug 28, 2018 @author: haidong ''' import unittest from sources.lastdigitoffibsum import getLastDigit class Test(unittest.TestCase): def testName(self): self.assertEqual(0, getLastDigit(0)) self.assertEqual(1, getLastDigit(1)) self.assertEqual(0, getLastDigit(6)) self.assertEqual(8, getLastDigit(9)) self.assertEqual(5, getLastDigit(100)) if __name__ == "__main__": #import sys;sys.argv = ['', 'Test.testName'] unittest.main()