Add type annotations

[Maddy Guthridge]

It'd be very useful if a few type annotations were added to the project, particularly for some of the methods used within tests.

This would make it much easier to write code using the library more quickly without needing to hunt through the documentation.