Remove py.test dependency, using plain unittest