def test_init(self): ''' test the initialization of investment class, ie. the assignment of position and num_trials ''' self.assertEqual( Investment.investment([1, 10, 100, 1000], 100).positions, [1, 10, 100, 1000]) self.assertEqual( Investment.investment([1, 10, 100, 1000], 100).num_trials, 100)
def test_investment_return(self): ''' test the investment return (daily) calculation function ''' investment=Investment.investment(10,1000) daily_return=investment.investment_return(10) self.assertEqual(len(daily_return),1000) self.assertGreaterEqual(daily_return.all(), -1) self.assertLessEqual(daily_return.all(), 1)
def test_investment_return(self): ''' test the investment return (daily) calculation function ''' investment = Investment.investment(10, 1000) daily_return = investment.investment_return(10) self.assertEqual(len(daily_return), 1000) self.assertGreaterEqual(daily_return.all(), -1) self.assertLessEqual(daily_return.all(), 1)
def test_init(self): ''' test the initialization of investment class, ie. the assignment of position and num_trials ''' self.assertEqual(Investment.investment([1,10,100,1000],100).positions,[1,10,100,1000]) self.assertEqual(Investment.investment([1,10,100,1000],100).num_trials,100)
'Invalid Input: all numbers of shares need to be positive') break except ValueError as msg: print(msg) except KeyboardInterrupt: print('Keyboard Interruption') sys.exit() except EOFError: print('EOFError') sys.exit() while True: try: inputstr2 = input( 'Please enter the number of times you want to repeat the simulation' ) if (inputstr2 == 'quit'): print("User Directed Quit.") sys.exit() num_trials = int(inputstr2) Investment.investment(positions, num_trials).simulation() break except ValueError as msg: print(msg) except KeyboardInterrupt: print('Keyboard Interruption') sys.exit() except EOFError: print('EOFError') sys.exit()