Add screening to stress tensor module

[marco-2023]

Checklist

• [x] Write a good description of what the PR does.
• [x] Add tests for each unit of code added (e.g. function, class)
• [x] Update documentation
• [x] Squash commits that can be grouped together
• [x] Rebase onto master

Type of Changes

	Type
✓	:sparkles: New feature

Related