Modified tests to not plot data.

Tests are run by py.test, and should not generate plots.