[testing] clean-up test_unary_ufuncs.py (#57615)
Summary:
Some clean-ups
Pull Request resolved: https://github.com/pytorch/pytorch/pull/57615
Reviewed By: albanD
Differential Revision: D28249173
Pulled By: mruberry
fbshipit-source-id: 19a300f6aa267932a7a92c2f5f377488f69bd822
