Artifact ROI

from random import random

def trial(winrate):
        losses = 0
        wins = 0
        while (losses < 2 and wins < 5):
                if (random() < winrate):
                        wins += 1
                else:
                        losses += 1

        return wins

tickets = {
        0: 0,
        1: 0,
        2: 0,
        3: 1,
        4: 1,
        5: 1
}

packs = {
        0: 0,
        1: 0,
        2: 0,
        3: 0,
        4: 1,
        5: 2
}

wins = [0] * 6
TRIALS = 100000
for i in range(TRIALS):
        wins[trial(0.515)] += 1

print("Tickets won: " + str(sum(wins[i] * tickets[i] for i in range(len(wins)))/TRIALS))
print("Packs won: " + str(sum(wins[i] * packs[i] for i in range(len(wins)))/TRIALS))