[MODELS] Add some tests