Adding tests to pip repo