test_util.py 文件源码

def paired_files():

    root = tempfile.mkdtemp()

    files = [[root, 'file1.mp3'], [root, 'file1.jams'],
             [root, 'file2.ogg'], [root, 'file2.jams'],
             [root, 'file3.wav'], [root, 'file3.jams'],
             [root, 'file4.flac'], [root, 'file4.jams']]

    files = [os.sep.join(_) for _ in files]

    for fname in files:
        with open(fname, 'w'):
            pass

    yield root, files

    for fname in files:
        try:
            os.remove(fname)
        except FileNotFoundError:
            pass

    os.rmdir(root)