added py.test and tests