Add the .py script to run the tests and generate the plot.
Add the .py script to run the tests and generate the plot.