Add test cases for training user database