def defaultSimulator(self):
# According to HWK 4 Example 1
balance = 1000000
startDate = dt.date(2011, 1, 10)
endDate = dt.date(2011, 12, 20)
orderFilename = "scratches/orders.csv"
historyFilename = "scratches/history.csv"
orders = OrderDf()
orders.fetchFromCsv(orderFilename)
symbols = orders.extractSymbols()
series = TimeSeries(startDate, endDate)
window = MarketWindow(series, symbols)
account = Account(balance, window)
history = PricePanel(window, self.provider)
sim = Simulator()
sim.config(account, window, orders, history)
sim.setBidBarPrice(ADJ_CLOSE)
return sim
def testOrders(self):
orderFilename = "scratches/orders.csv"
orders = OrderDf()
orders.fetchFromCsv(orderFilename)
symbols = orders.extractSymbols()
self.assertEqual(len(symbols), 4)
self.assertEqual(len(orders), 14)
