Update unit test
Add AModel unit tests