move tests to unittest format