110 lines
3.3 KiB
Python
110 lines
3.3 KiB
Python
""" test FileUtil """
|
|
|
|
import os
|
|
import pytest
|
|
|
|
TEST_HEIC = "tests/test-images/IMG_3092.heic"
|
|
TEST_RAW = "tests/test-images/DSC03584.dng"
|
|
|
|
|
|
def test_copy_file_valid():
|
|
# copy file with valid src, dest
|
|
import os.path
|
|
import tempfile
|
|
from osxphotos.fileutil import FileUtil
|
|
|
|
temp_dir = tempfile.TemporaryDirectory(prefix="osxphotos_")
|
|
src = "tests/test-images/wedding.jpg"
|
|
result = FileUtil.copy(src, temp_dir.name)
|
|
assert result == 0
|
|
assert os.path.isfile(os.path.join(temp_dir.name, "wedding.jpg"))
|
|
|
|
|
|
def test_copy_file_invalid():
|
|
# copy file with invalid src
|
|
import tempfile
|
|
from osxphotos.fileutil import FileUtil
|
|
|
|
temp_dir = tempfile.TemporaryDirectory(prefix="osxphotos_")
|
|
with pytest.raises(Exception) as e:
|
|
src = "tests/test-images/wedding_DOES_NOT_EXIST.jpg"
|
|
assert FileUtil.copy(src, temp_dir.name)
|
|
assert e.type == FileNotFoundError
|
|
|
|
|
|
def test_copy_file_norsrc():
|
|
# copy file with --norsrc
|
|
import os.path
|
|
import tempfile
|
|
from osxphotos.fileutil import FileUtil
|
|
|
|
temp_dir = tempfile.TemporaryDirectory(prefix="osxphotos_")
|
|
src = "tests/test-images/wedding.jpg"
|
|
result = FileUtil.copy(src, temp_dir.name, norsrc=True)
|
|
assert result == 0
|
|
assert os.path.isfile(os.path.join(temp_dir.name, "wedding.jpg"))
|
|
|
|
|
|
def test_hardlink_file_valid():
|
|
# hardlink file with valid src, dest
|
|
import os.path
|
|
import tempfile
|
|
from osxphotos.fileutil import FileUtil
|
|
|
|
temp_dir = tempfile.TemporaryDirectory(prefix="osxphotos_")
|
|
src = "tests/test-images/wedding.jpg"
|
|
dest = os.path.join(temp_dir.name, "wedding.jpg")
|
|
FileUtil.hardlink(src, dest)
|
|
assert os.path.isfile(dest)
|
|
assert os.path.samefile(src, dest)
|
|
|
|
|
|
def test_unlink_file():
|
|
import os.path
|
|
import tempfile
|
|
from osxphotos.fileutil import FileUtil
|
|
|
|
temp_dir = tempfile.TemporaryDirectory(prefix="osxphotos_")
|
|
src = "tests/test-images/wedding.jpg"
|
|
dest = os.path.join(temp_dir.name, "wedding.jpg")
|
|
result = FileUtil.copy(src, temp_dir.name)
|
|
assert os.path.isfile(dest)
|
|
FileUtil.unlink(dest)
|
|
assert not os.path.isfile(dest)
|
|
|
|
|
|
@pytest.mark.skipif(
|
|
"OSXPHOTOS_TEST_CONVERT" not in os.environ,
|
|
reason="Skip if running in Github actions, no GPU.",
|
|
)
|
|
def test_convert_to_jpeg():
|
|
""" test convert_to_jpeg """
|
|
import pathlib
|
|
import tempfile
|
|
from osxphotos.fileutil import FileUtil
|
|
|
|
temp_dir = tempfile.TemporaryDirectory(prefix="osxphotos_")
|
|
with temp_dir:
|
|
imgfile = pathlib.Path(TEST_HEIC)
|
|
outfile = pathlib.Path(temp_dir.name) / f"{imgfile.stem}.jpeg"
|
|
assert FileUtil.convert_to_jpeg(imgfile, outfile)
|
|
assert outfile.is_file()
|
|
|
|
@pytest.mark.skipif(
|
|
"OSXPHOTOS_TEST_CONVERT" not in os.environ,
|
|
reason="Skip if running in Github actions, no GPU.",
|
|
)
|
|
def test_convert_to_jpeg_quality():
|
|
""" test convert_to_jpeg with compression_quality """
|
|
import pathlib
|
|
import tempfile
|
|
from osxphotos.fileutil import FileUtil
|
|
|
|
temp_dir = tempfile.TemporaryDirectory(prefix="osxphotos_")
|
|
with temp_dir:
|
|
imgfile = pathlib.Path(TEST_RAW)
|
|
outfile = pathlib.Path(temp_dir.name) / f"{imgfile.stem}.jpeg"
|
|
assert FileUtil.convert_to_jpeg(imgfile, outfile, compression_quality=0.1)
|
|
assert outfile.is_file()
|
|
assert outfile.stat().st_size < 1000000
|