[xla:cpu] Restore the verbose version of custom call tests

Reverts 71f2fdb. Applies necessary changes to the new tests.

Rollback reason: The more verbose version provides more context and is self-documenting; thus, it is easier to understand.

Reverts 8a2d316

PiperOrigin-RevId: 625249016