Add tests, homogenise and document functions of utils for inference model

Add tests, homogenise and document functions of utils for inference model #295