Added tests to context

Doesn't actually work. Not sure why.