add assert data function
fix test case