Improve test coverage

[djbower]

The test coverage is generally low, in part because much of the functionality involves plotting which is intrinsically more difficult to test than functions that return numerical values.