def testCountOnTrack(self):
count0 = 0
count1 = 0
for i in range(1, 100):
train = TrainTrack(i // 2, i)
for j in range(1, 100):
list = train.simulate(j)[1:]
count0 += list.count(0)
count1 += list.count(1)
answer = count0 < count1
self.assertEqual(True, answer)
def simulateTrains(L0, L1, numTrains, duration):
tracks = [TrainTrack(L0, L1) for x in range(numTrains)]
results = [x.simulate(duration) for x in tracks]
# print(results)
fig, ax = plt.subplots(numTrains)
fig.suptitle("Presence of Trains on Tracks every second")
for i in range(len(results)):
ax[i].plot([x for x in range(duration)], results[i])
ax[i].set_title("Track "+str(i))
plt.show()
def simulateGame(L0, L1, numTrains, lengthOfGame, hoboSmartness, trackResults=[]):
def showGraphs():
fig, ax = plt.subplots(numTrains)
fig.suptitle("Presence of Trains on Tracks every second")
deathTrack = 0
for i in range(len(trackResults)):
hoboHistoryOnThisTrack = []
for j in range(len(dumbledore.positionHistory)):
if dumbledore.positionHistory[j] == i:
# # print("J: "+str(j))
hoboHistoryOnThisTrack.append(.5)
if j == score:
# print("SCORE: "+str(score))
# print("DEATH TRACK: "+str(i))
deathTrack = i
else:
hoboHistoryOnThisTrack.append(None)
ax[i].plot([x for x in range(lengthOfGame)], trackResults[i], hoboHistoryOnThisTrack, 'ro')
ax[i].set_title("Track " + str(i))
ax[deathTrack].text(score, .5, "HOBO DIED", bbox=dict(facecolor='red', alpha=0.5))
plt.show()
# for each 60second window generate TrainTrack results
tracks = [TrainTrack(L0, L1) for x in range(numTrains)]
if trackResults == []:
trackResults = [x.simulate(lengthOfGame) for x in tracks]
# print("TRACKRESULTS: "+str(trackResults))
dumbledore = Hobo()
dumbledore.runningResults = [[0] for x in range(numTrains)]
score = lengthOfGame
for i in range(lengthOfGame):
# DID HE GET HIT?
if trackResults[dumbledore.position][i] == 1:
dumbledore.hp -= 1
# O SHIT, HE DED END THE GAME
if dumbledore.hp == 0:
score = i
# showGraphs()
return score
# GET PAPER PLANE
paperPlane = []
if (random.randint(1, 100) <= 75) and i < lengthOfGame - 1:
for y in range(numTrains):
paperPlane.append(trackResults[y][i+1])
else:
for y in range(numTrains):
paperPlane.append(random.randint(0, 1))
dumbledore.getSuggestion(paperPlane)
# LOOKS AROUND AT TRAINS
dumbledore.lookAtTracks(i, trackResults)
# STORES WHAT HE SEES
dumbledore.updateResults()
# JUMP
dumbledore.act(hoboSmartness)
# showGraphs()
return score
def simulateGame(L0, L1, numTrains, lengthOfGame, hoboSmartness):
def showGraphs():
fig, ax = plt.subplots(numTrains)
fig.suptitle("Presence of Trains on Tracks every second")
deathTrack = 0
for i in range(len(trackResults)):
hoboHistoryOnThisTrack = []
for j in range(len(dumbledore.positionHistory)):
if dumbledore.positionHistory[j] == i:
hoboHistoryOnThisTrack.append(.5)
if j == score:
deathTrack = i
else:
hoboHistoryOnThisTrack.append(None)
ax[i].plot([x for x in range(lengthOfGame)], trackResults[i],
hoboHistoryOnThisTrack, 'ro')
ax[i].set_title("Track " + str(i))
ax[deathTrack].text(score,
.5,
"HOBO DIED",
bbox=dict(facecolor='red', alpha=0.5))
plt.show()
# for each time period generate a trainTrack simulation
tracks = [TrainTrack(L0, L1) for x in range(numTrains)]
trackResults = [x.simulate(lengthOfGame) for x in tracks]
# initialize our hobo object
dumbledore = Hobo()
dumbledore.runningResults = [[0] for x in range(numTrains)]
score = lengthOfGame
# MAIN GAME LOOP, each i represents a game slice or a second of the whole window
for i in range(lengthOfGame):
# DID HE GET HIT? Does the track he is on have a 1 at the given index? If so, decrement HP
if trackResults[dumbledore.position][i] == 1:
dumbledore.hp -= 1
# O SHIT, HE DIED END THE GAME
if dumbledore.hp == 0:
score = i
return score
# GET PAPER PLANE
paperPlane = []
if (random.randint(1, 100) <= 99) and i < lengthOfGame - 1:
for y in range(numTrains):
paperPlane.append(trackResults[y][i + 1])
else:
for y in range(numTrains):
paperPlane.append(random.randint(0, 1))
dumbledore.getSuggestion(paperPlane)
def testLength1(self):
train = TrainTrack(1, 1)
value = train.simulate(1)
self.assertEqual([0], value)
def testZeroL1(self):
train = TrainTrack(10, 0)
value = 1 in train.simulate(20)[1:]
self.assertEqual(True, value)
def testZeroL0(self):
train = TrainTrack(0, 10)
value = 0 in train.simulate(20)[1:]
self.assertEqual(False, value)
def testLength1000(self):
train = TrainTrack(10, 10)
answer = len(train.simulate(1000))
self.assertEqual(1000, answer)