As mentioned in 4. of this comment, there are several plotting functions that could use unit tests in plotting.py. At the moment the plotting test doesn't properly "test" anything, they primarily flush the code paths to make sure there aren't any errors raised. But adding some unit tests would be a good start.