[WIP] Extract test functionality

Fix #101.
sounds good, yes.