How to plot built-in functions of pybamm before solving a model?

[zhengq01]

Hi there, I wonder how to visualize the built-in functions, such as pybamm.EqualHeaviside(t, t_end), without solving models and using sim.plot(). I want to plot the function before solving models. How can I do that? Thanks a lot!

[TomTranter]

Something like this

import pybamm
import numpy as np

a = pybamm.Scalar(np.linspace(0, 1, 4))
b = pybamm.Scalar(0.5)
c = pybamm.EqualHeaviside(a, b).to_equation()
print(a, b, c)

[rtimms]

import pybamm

x = pybamm.linspace(0, 1, 100)
y = pybamm.EqualHeaviside(x, 0.5)  # this is the expression 
y_eval = pybamm.Array(y.evaluate())  # this is the evaluated expression
pybamm.plot(x, y_eval) 

[zhengq01]

Thanks a lot! It works.