More tests, remove plotting from tests

1 job for energy_tests in 3 minutes and 59 seconds