* chore: improve mypy Signed-off-by: Henry Schreiner <[email protected]> * chore(types): type functions in tests Signed-off-by: Henry Schreiner <[email protected]> * chore(types): No partial types in tests Signed-off-by: Henry Schreiner <[email protected]> --------- Signed-off-by: Henry Schreiner <[email protected]>