def run_test(self, log, test):
try:
# https://stackoverflow.com/questions/16710076/python-split-a-string-respect-and-preserve-quotes
args = re.findall(r'(?:[^\s,"]|"(?:\\.|[^"])*")+', test["options"])
options, args = self.op.parse_args(args)
simulator = Simulator(options, self.RDTImpl)
if options.capture_log:
sys.stdout = log
result = simulator.Simulate()
return self.check_test_results(test, simulator, result)
except Exception as e:
return False, e
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_()
p.seqnum = packet.seqnum
#p.checksum = sum
simulator.to_layer3(EventEntity.B, p)
else:
print("ERROR: B received a corrupt packet!")
p.acknum = 0 if packet.seqnum == 1 else 1
p.seqnum = ""
simulator.to_layer3(EventEntity.B, p)
def checksum(packet):
checksum = 0
for i in range (0, len(packet.payload)):
checksum = (checksum + ord(packet.payload[i]))
checksum = int(packet.seqnum) + packet.acknum + checksum
return checksum
if __name__ == "__main__":
rdt = ReliableDataTransferProtocol()
simulator = Simulator(rdt)
simulator.Setup()
simulator.Simulate()
#rdt.A_Init()
#rdt.B_Init()
#!/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
