[jit] Updated tests checking save and load transforms (#6248)

* Updated tests to save and load transforms

* Fixed code formatting issue

Author: vfdev-5