Set test_energy