[Feature] Add Mypy to tests

# Is your feature request related to a problem? Please describe.

Static typing can be useful to track down bugs and check for problematic behavior. There should be tests that use Mypy to check that all types are correct.

# Describe the feature you'd like.

Add Mypy to tests
