def test():
'''
Unit test for Simulator
'''
from Simulator import Simulator
from Strategy import SimpleStrategy
from Market import Market
from User import User
# from Gui import Gui
from PriceSource import PriceSource
import datetime
from PyQt4 import QtGui
app = QtGui.QApplication([])
strategy = SimpleStrategy()
name = 'Weichao Qiu'
initial_money = 10000
user = User(name, initial_money, strategy)
# sz01 = PriceSource('000001.SZ')
market = Market()
# market.AddPriceSource(sz01) # Done by ctor of Market class
market.AddUser(user)
simulator = Simulator(market)
# simulator.gui = Gui()
simulator.start_date = datetime.datetime(2011, 1, 10)
simulator.end_date = datetime.datetime(2011, 3, 15)
# simulator.end_date = simulator.start_date + datetime.timedelta(180)
# simulate for 6 months
# simulator.end_date = datetime.datetime.now()
simulator.Simulate() # Begin simulation
print 'end of simulation'
app.exec_()
#!/usr/bin/python
from Simulator import Simulator
from Strategy import SimpleStrategy
from Market import Market
from User import User
from Gui import Gui
from PriceSource import PriceSource
import datetime
strategy = SimpleStrategy()
name = 'Weichao Qiu'
initial_money = 10000
user = User(name, initial_money, strategy)
sz01 = PriceSource('000001.SZ')
market = Market()
market.AddPriceSource(sz01)
market.AddUser(user)
simulator = Simulator(market)
# simulator.gui = Gui()
simulator.start_date = datetime.datetime(2011, 1, 10)
simulator.Simulate() # Begin simulation
