[WIP] Extract test functionality

Fix #101.
agreed.