Allow test to take heap snapshot at custom point

[xleclercq]

Typically this is done before and after the test but if the test has some particular needs it should be possible to change that default.