update the unittest
2 files changed