import pybamm
import matplotlib.pyplot as plt

options_list = [
    ("two-particle", { "particle phases": ("1", "1")}),
    ("algebraic", {"surface form": "algebraic"}),
    ("standard", {})
]

sims = []
for label, options in options_list:
    model = pybamm.lithium_ion.SPMe(options)
    sim = pybamm.Simulation(model)
    sim.solve([0, 3600])
    sims.append((label, sim))

fig, ax = plt.subplots(1, 1)
for label, sim in sims:
    ax.plot(
        sim.solution["Time [min]"].data,
        sim.solution["Battery voltage [V]"].data,
        marker= 'o' if label == 'algebraic' else 's' if label == 'two-particle' else '^',
        label=label
    )

ax.set_xlabel("Time [min]")
ax.set_ylabel("Voltage [V]")
ax.legend()
plt.show()