added unittest for model