def main():
# Generate fake trades for each stock
trade_stocks = []
for ss in SAMPLE:
trade_stocks.append(Stock(ss))
for trade_stock in trade_stocks:
trade_stock.trades = []
fake_trades = generate_trades()
for fake_trade in fake_trades:
trade_stock.record_trade(
fake_trade[0], fake_trade[1], fake_trade[2], fake_trade[3]
)
# Calculate the stock price based on trades in the last 15 minutes
# Calculate the All Share Index
since = datetime.now() + timedelta(minutes=-15)
prices = []
for trade_stock in trade_stocks:
print 'Stock price for %s: %s' % (
trade_stock.symbol,
trade_stock.calculate_stock_price(since)
)
prices.extend(trade_stock.all_prices())
print 'GBCE All Share Index: %s' % geometric_mean(prices)
def test_geometric_mean_success(self):
prices = [2, 8]
self.assertEqual(
geometric_mean(prices),
4
)