Changed tests to PyUnit