def test_standard_filepath(config_class, fmt):
filepath = standard_filepath(config=config_class, rootname="root", fmt=fmt)
filetypes = list(guess(filepath))
assert len(filetypes) == 1
filetype = filetypes[0]
assert filetype.filepath == filepath
assert filetype.fmt == fmt
assert filetype.config_class == config_class
filepath2 = standard_filepath(config=config_class.__name__, rootname="root", fmt=fmt)
assert filepath2 == filepath
filepath3 = standard_filepath(config=config_class(), rootname="root", fmt=fmt)
assert filepath3 == filepath
def test_standard_filepath(config_class, protocol):
filepath = standard_filepath(config=config_class, rootname="root", protocol=protocol)
filetypes = list(guess(filepath))
assert len(filetypes) == 1
filetype = filetypes[0]
assert filetype.filepath == filepath
assert filetype.protocol == protocol
assert filetype.config_class == config_class
filepath2 = standard_filepath(config=config_class.__name__, rootname="root", protocol=protocol)
assert filepath2 == filepath
filepath3 = standard_filepath(config=config_class(), rootname="root", protocol=protocol)
assert filepath3 == filepath
def test_classify(tmpdir):
config_classes = get_config_classes()
protocols = get_protocols()
ref_filetypes = {}
for rootname in 'x', 'xy':
for config_class in config_classes:
for protocol in protocols:
filename = standard_filepath(rootname=rootname, config=config_class, protocol=protocol)
filepath = os.path.join(tmpdir.strpath, filename)
with open(filepath, "w") as f_out:
pass
ref_filetype = FileType(filepath=filepath, config_class=config_class, protocol=protocol)
ref_filetypes[filepath] = ref_filetype
for filetype in classify(tmpdir.strpath):
assert filetype.filepath in ref_filetypes
ref_filetype = ref_filetypes.pop(filetype.filepath)
assert filetype == ref_filetype
os.remove(filetype.filepath)
assert len(ref_filetypes) == 0
def test_classify(tmpdir):
config_classes = get_config_classes()
fmts = get_fmts()
ref_filetypes = {}
rootnames = ('x', 'xy')
for rootname in rootnames:
for config_class in config_classes:
config_class_name = get_config_class_name(config_class)
for fmt in fmts:
filename = standard_filepath(rootname=rootname, config=config_class_name, fmt=fmt)
filepath = os.path.join(tmpdir.strpath, filename)
with open(filepath, "w") as f_out:
pass
ref_filetype = FileType(filepath=filepath, config_class=config_class, fmt=fmt)
ref_filetypes[filepath] = ref_filetype
assert len(ref_filetypes) == len(rootnames) * len(config_classes) * len(fmts)
for filetype in classify(tmpdir.strpath):
assert filetype.filepath in ref_filetypes
ref_filetype = ref_filetypes.pop(filetype.filepath)
assert filetype == ref_filetype
os.remove(filetype.filepath)
assert len(ref_filetypes) == 0
