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)