facefusion/tests/test_memory.py

from facefusion.common_helper import is_linux, is_macos
from facefusion.memory import limit_system_memory


def test_limit_system_memory() -> None:
	assert limit_system_memory(4) is True
	if is_linux() or is_macos():
		assert limit_system_memory(1024) is False