class TestFileBag(unittest.TestCase): def setUp(self): self.file_bag = FileBag('testfilebag', use_tempdir=True) def test__mk_file(self): f = self.file_bag.mk_file(suffix='.bar', prefix='foo') f.write('blah') f.close() def test__mk_file2(self): (f1, f2) = self.file_bag.mk_two_files(suffix1='.bar1', suffix2='.bar2', prefix='foo') f1.write('blah1') f1.close() f2.write('blah2') f2.close()