fix __all__ and start writing test