def test_check_memory_allocation_env(monkeypatch, mock_get_available_memory,
allowed_env, allowed_explicit, result):
"""Check environmental control over memory check"""
if allowed_env is None:
monkeypatch.delenv('DMODEL_ALLOWED_MEMORY', raising=False)
else:
monkeypatch.setenv('DMODEL_ALLOWED_MEMORY', allowed_env)
# Allocate amount that would fit at 100% + swap.
can_allocate, required = util.check_memory_allocation(
(MEMORY // 2, 1),
allowed=allowed_explicit,
)
assert can_allocate is result
def test_check_memory_allocation(mock_get_available_memory, dim, allowed,
include_swap, result):
"""Check general operation of check_memory_allocation"""
can_allocate, required = util.check_memory_allocation(
(dim, 1), allowed=allowed, include_swap=include_swap)
assert can_allocate is result