Support py.test framework