Add unit test for logical model