Move context tests to their own module.

Closes issue #218