[Tests] Fix Python docstring.