add data model test code