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