diff --git a/src/grass_gis_helpers/general.py b/src/grass_gis_helpers/general.py index 2403f4e..7df9d5c 100644 --- a/src/grass_gis_helpers/general.py +++ b/src/grass_gis_helpers/general.py @@ -111,3 +111,23 @@ def free_ram(unit, percent=100): return int(round(memory_gb_percent)) else: grass.fatal(f"Memory unit {unit} not supported") + + + +def test_memory(memory_string): + """Function returns available memory. + Args: + memory_string(string): string from standard memory input option + Returns: + memory(int): available memory to use""" + # check memory + memory = int(memory_string) + ram = free_ram("MB", 100) + if ram < memory: + grass.warning( + _(f"Using {memory} MB but only {ram} MB RAM available.") + ) + grass.warning(_(f"Set used memory to {ram} MB.")) + return ram + else: + return memory \ No newline at end of file