def test_ah_avg_price(self): # method test_ah_avg_price checks if the average value # returned from ah_avg is greater than 0 stock_symbol = 'yhoo' test = after_hours.ah_avg(stock_symbol) test = test[1] self.assertGreater(test, 0, msg='Test ah_avg price failed')
def test_ah_avg_symbol(self): # method ah_avg_symbol checks if the symbol value # from ah_avg is being returned accurately stock_symbol = 'yhoo' test = after_hours.ah_avg(stock_symbol) test = test[0] self.assertEqual(test, 'yhoo', msg='Test ah_avg symbol failed')