import a1

import json, pprint, random

# Enums, never heard of her
class PRIVOUS_TASK: pass
class EXACT: pass

TESTS = (
    (
        3, # Task
        5, # Number of tests
        a1.determine_cost_to_pump, # The function to test
        ( # The args and their ranges
            ("gen_power", (float, 100, 600, 50)), # Type, min, max, inc
            ("pumping_time", (float, 1, 10, 1)), 
            ("off_peak_tariff", (float, 0.01, 0.1, 0.01)),
        )
    ),
)

for task, iterations, function, args in TESTS:
    print(f'\nTask {task}')
    
    args_used = []
    results = []
    for i in range(iterations):
        this_arg = {}
        for arg, vals in args:
            this_arg[arg] = random.randrange(vals[1], vals[2], vals[3])

        args_used.append(this_arg)
        results.append(function(*args_used[-1]))

    pprint(args_used)
    pprint(results)