Add missing tests and forgotten param in docstring.