redefine the test with unittest