refactoring: move IR unit tests to separate files.

Previously,  unit tests and code lived in the same '*.rs' file.
We have written too much tests and these files became bloated
(up to 1000 lines) - it is difficult to navigate them. So, lets
make a refactoring.