Improve unit test for model loading method

Change-Id: I3a97f2f87e8d2a3309cc9ca1d1895dbdfc4ed047