feat: current context for test classes

[grigoriev]

Proposed changes

Describe the use case and detail of the change. If this PR addresses an issue on GitHub, make sure to include a link to that issue using one of the supported keywords here in this description (not in the title of the PR).

Checklist

Before creating a PR, run through this checklist and mark each as complete:

• [x] I have read the CONTRIBUTING document
• [x] If applicable, I have added tests that prove my fix is effective or that my feature works
• [x] If applicable, I have checked that any relevant tests pass after adding my changes
• [ ] I have updated any relevant documentation