class AppTestFile(BaseROTests): expected_filename = str(udir.join('sample')) expected_mode = 'rb' extra_args = () spaceconfig = {'usemodules': ['binascii', 'time', 'struct']} def setup_method(self, method): space = self.space if hasattr(space, 'gettypeobject'): from pypy.module._file.interp_file import W_File w_filetype = space.gettypeobject(W_File.typedef) else: w_filetype = file # TinyObjSpace, for "py.test -A" self.w_file = space.call_function( w_filetype, space.wrap(self.expected_filename), space.wrap(self.expected_mode), *[space.wrap(a) for a in self.extra_args]) self.w_sample = space.wrap(self.sample) self.w_expected_filename = space.wrap(self.expected_filename) self.w_expected_mode = space.wrap(self.expected_mode) self.w_expected_lines = space.wrap(self.get_expected_lines()) def teardown_method(self, method): self.space.call_method(self.w_file, 'close')
class TestWithCPython(BaseROTests): expected_filename = str(udir.join('sample')) expected_mode = 'rb' def setup_method(self, method): self.file = open(self.expected_filename, self.expected_mode) self.expected_lines = self.get_expected_lines() def teardown_method(self, method): self.file.close()
def setup_method(self, method): fn = str(udir.join('temptestfile')) self.w_temptestfile = self.space.wrap(fn)
def setup_module(mod): udir.join('sample').write(SAMPLE, 'wb')