Add unit tests for MOSES wrapper