Adds context tests.

Resolves #53