add unittest
 .