import highspy
import timeit

h = highspy.Highs()

h.readModel("80bau3b.mps")
h.setOptionValue("time_limit", 10)

# Solve
start_time = timeit.default_timer()
h.run()
elapsed = timeit.default_timer() - start_time

print(f"Time taken to solve: {elapsed:.4f}")

solution = h.getSolution()
num_vars = h.getNumCol()

# Extract values using __getitem__ N times
start_time = timeit.default_timer()
values = [solution.col_value[icol] for icol in range(num_vars)]
elapsed = timeit.default_timer() - start_time
print(f"__getitem__ method for {num_vars} vars: {elapsed:.4f}")


# Extract values by copying first to list
start_time = timeit.default_timer()
col_values = list(solution.col_value)
values = [col_values[icol] for icol in range(num_vars)]
elapsed = timeit.default_timer() - start_time
print(f"Copy-first method for {num_vars} vars: {elapsed:.4f}")