def test_all_share_index_realistic(self):
# More realistic case, spread of stock values
stocks = {
"100": self.stock_100,
"105": self.stock_105,
"110": self.stock_110,
"200": self.stock_200
}
exchange = Exchange("TESTEX", stocks)
all_share_index = exchange.calculate_all_share_index()
self.assertAlmostEqual(all_share_index, 123.283,
places=3) # note rounding.
def test_all_share_index_empty(self):
# Edge case, no stocks
exchange = Exchange("TESTEX", {})
all_share_index = exchange.calculate_all_share_index()
self.assertEqual(all_share_index, 0)
def test_all_share_index_basic(self):
# basic test case, 1 stock of value 100, means the index value is 100
exchange = Exchange("TESTEX", {"100": self.stock_100})
all_share_index = exchange.calculate_all_share_index()
self.assertEqual(all_share_index, 100)
