Add more tests with unittest.