add unittest for math