switch tests to unittest