Added some tests for Memory
Returning correct context.