Extract test functionality

Fix #101.
do you repeat this from the base class?