Update tests for new magic_arguments docstrings