wheel: improve unit tests