simulator.py

from coin import Coin
from basic_models import *
from calm_models import *
from indecisive_models import *


def simulation(model, prompt=False):
    score = 0
    while model.fund > 0:
        coin = Coin()
        model.test(coin)
        is_correct = model.reward(coin)

        if is_correct:
            score += 1

        if prompt:
            print(f"Score: {score}")
            print(f"Remaining: {model.fund}")
    return score


def test_model(model, num_test=100, prompt_records=False):
    global FUND
    records = []
    for _ in range(num_test):
        records.append(simulation(model))
        model.fund = FUND
    if prompt_records:
        print(records)
    return f"Min: {min(records)}, Mean: {sum(records) / len(records)}, Max: {max(records)}"


# Simulation
FUND = 100

# Basic Models
none_belief_model = NoBeliefModel(FUND)
weak_belief_model = WeakBeliefModel(FUND)
fanatic_model = FanaticModel(FUND)
belief_model = BeliefModel(FUND)

print(f"Basic models #########################")
print(f"No belief model Result: {test_model(none_belief_model, 10000)}")
print(f"Weak belief model Result: {test_model(weak_belief_model, 10000)}")
print(f"Fanatic model Result: {test_model(fanatic_model, 10000)}")
print(f"Belief model Result: {test_model(belief_model, 10000)}")

# Calm Models
calm_weak_belief_model = CalmWeakBeliefModel(FUND)
calm_fanatic_model = CalmFanaticModel(FUND)
calm_belief_model = CalmBeliefModel(FUND)

print(f"Calm models ###########################")
print(f"Calm weak belief model Result: {test_model(calm_belief_model, 10000)}")
print(f"Calm fanatic model Result: {test_model(calm_fanatic_model, 10000)}")
print(f"Calm belief model Result: {test_model(calm_belief_model, 10000)}")

# Indecisive Models
basic_indecisive_model = BasicIndecisiveModel(FUND, significance_level=0.3)
indecisive_weak_belief_model = IndecisiveWeakBeliefModel(FUND, significance_level=0.3)
indecisive_fanatic_model = IndecisiveFanaticModel(FUND, significance_level=0.3)
indecisive_belief_model = IndecisiveBeliefModel(FUND, significance_level=0.3)

print(f"Indecisive models #####################")
print(f"Basic indecisive model Result: {test_model(basic_indecisive_model, 10000)}")
print(f"Indecisive weak belief model Result: {test_model(indecisive_weak_belief_model, 10000)}")
print(f"Indecisive fanatic model Result: {test_model(indecisive_fanatic_model, 10000)}")
print(f"Indecisive belief model Result: {test_model(indecisive_belief_model, 10000)}")