def test_n_days_return(self): ''' Test that n_days_returns() generates the correct list of values ''' position_1000 = InvestmentPositions(1, 1000) self.assertEqual(position_1000.n_days_return(10), [2000, 0, 0, 2000, 2000, 0, 2000, 2000, 0, 0])
def test_n_days_return(self): ''' Test that n_days_returns() generates the correct list of values ''' position_1000 = InvestmentPositions(1, 1000) self.assertEqual(position_1000.n_days_return(10), [2000, 0, 0, 2000, 2000, 0, 2000, 2000, 0, 0])