Add tests for context manager behavrior of Context.push
Removed leftover import