def test_get_compression(): assert _get_compressor("lz4") == numcodecs.LZ4(numcodecs.lz4.DEFAULT_ACCELERATION) assert _get_compressor(None) is None assert _get_compressor("default") == "default" assert _get_compressor("zstd") == numcodecs.Zstd(numcodecs.zstd.DEFAULT_CLEVEL) assert _get_compressor("png") == PngCodec(solo_channel=True) with pytest.raises(ValueError): _get_compressor("abcd")
def _get_compressor(self, compressor: str): if compressor.lower() == "lz4": return numcodecs.LZ4(numcodecs.lz4.DEFAULT_ACCELERATION) elif compressor.lower() == "zstd": return numcodecs.Zstd(numcodecs.zstd.DEFAULT_CLEVEL) elif compressor.lower() == "default": return "default" else: raise ValueError( f"Wrong compressor: {compressor}, only LZ4 and ZSTD are supported" )
def _get_compressor(compressor: str): if compressor is None: return None elif compressor.lower() == "lz4": return numcodecs.LZ4(numcodecs.lz4.DEFAULT_ACCELERATION) elif compressor.lower() == "zstd": return numcodecs.Zstd(numcodecs.zstd.DEFAULT_CLEVEL) elif compressor.lower() == "default": return "default" elif compressor.lower() == "png": return PngCodec(solo_channel=True) else: raise ValueError( f"Wrong compressor: {compressor}, only LZ4, PNG and ZSTD are supported" )