增加对这两个类的测试代码