WIP tests

Remove this before landing.