add a test in `test_plot.py` for testing.