diff options
| -rw-r--r-- | tests/test_file_utils.py | 95 |
1 files changed, 75 insertions, 20 deletions
diff --git a/tests/test_file_utils.py b/tests/test_file_utils.py index f35999d..65ea0c4 100644 --- a/tests/test_file_utils.py +++ b/tests/test_file_utils.py @@ -3,31 +3,86 @@ import os from qolab.file_utils import infer_compression, save_table_with_header + def test_infer_compression(): - assert infer_compression('data.dat') is None - assert infer_compression('data.unknown_ext') is None - assert infer_compression('data.dat.gz') == 'gzip' - assert infer_compression('data.dat.bz') == 'bzip' - assert infer_compression('data.dat.bz2') == 'bzip' + assert infer_compression("data.dat") is None + assert infer_compression("data.unknown_ext") is None + assert infer_compression("data.dat.gz") == "gzip" + assert infer_compression("data.dat.bz") == "bzip" + assert infer_compression("data.dat.bz2") == "bzip" + def test_save_table_with_header_match_filename_to_compression(): import tempfile - data = [ [1, 3], [2, 5] ] - headerstr=['header 1', 'header 2'] - with tempfile.TemporaryDirectory(prefix='qolab_test') as d: - fname = os.path.join(d, 't.dat') - assert save_table_with_header(fname, data, header=headerstr, compressionmethod=None, match_filename_to_compression=True) == (fname) - assert save_table_with_header(fname, data, header=headerstr, compressionmethod='gzip', match_filename_to_compression=True) == (fname+'.gz') - assert save_table_with_header(fname, data, header=headerstr, compressionmethod='bzip', match_filename_to_compression=True) == (fname+'.bz') + + data = [[1, 3], [2, 5]] + headerstr = ["header 1", "header 2"] + with tempfile.TemporaryDirectory(prefix="qolab_test") as d: + fname = os.path.join(d, "t.dat") + assert save_table_with_header( + fname, + data, + header=headerstr, + compressionmethod=None, + match_filename_to_compression=True, + ) == (fname) + assert save_table_with_header( + fname, + data, + header=headerstr, + compressionmethod="gzip", + match_filename_to_compression=True, + ) == (fname + ".gz") + assert save_table_with_header( + fname, + data, + header=headerstr, + compressionmethod="bzip", + match_filename_to_compression=True, + ) == (fname + ".bz") # now cases when requested extension does not match the compression - assert save_table_with_header(fname+'.gz', data, header=headerstr, compressionmethod=None, match_filename_to_compression=True) == (fname+'.gz.dat') - assert save_table_with_header(fname+'.bz', data, header=headerstr, compressionmethod='gzip', match_filename_to_compression=True) == (fname+'.bz.gz') - assert save_table_with_header(fname+'.gz', data, header=headerstr, compressionmethod='bzip', match_filename_to_compression=True) == (fname+'.gz.bz') + assert save_table_with_header( + fname + ".gz", + data, + header=headerstr, + compressionmethod=None, + match_filename_to_compression=True, + ) == (fname + ".gz.dat") + assert save_table_with_header( + fname + ".bz", + data, + header=headerstr, + compressionmethod="gzip", + match_filename_to_compression=True, + ) == (fname + ".bz.gz") + assert save_table_with_header( + fname + ".gz", + data, + header=headerstr, + compressionmethod="bzip", + match_filename_to_compression=True, + ) == (fname + ".gz.bz") # do as I told cases, extension does not match compression - assert save_table_with_header(fname+'.gz', data, header=headerstr, compressionmethod=None, match_filename_to_compression=False) == (fname+'.gz') - assert save_table_with_header(fname, data, header=headerstr, compressionmethod='gzip', match_filename_to_compression=False) == (fname) - assert save_table_with_header(fname, data, header=headerstr, compressionmethod='bzip', match_filename_to_compression=False) == (fname) - - + assert save_table_with_header( + fname + ".gz", + data, + header=headerstr, + compressionmethod=None, + match_filename_to_compression=False, + ) == (fname + ".gz") + assert save_table_with_header( + fname, + data, + header=headerstr, + compressionmethod="gzip", + match_filename_to_compression=False, + ) == (fname) + assert save_table_with_header( + fname, + data, + header=headerstr, + compressionmethod="bzip", + match_filename_to_compression=False, + ) == (fname) |