test-driver: add mypy support

It's a good idea to expand this in future to test code as well,
so we get type checking there as well.