Fix units and update tests. (#12)

neuroinformatics-unit · Dec 18, 2023 · 6e2636e · 6e2636e
1 parent 20dfb03
commit 6e2636e
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/slurmio/slurmio.py b/slurmio/slurmio.py
@@ -166,6 +166,6 @@ def _trackable_resources(self) -> dict:
 
 def convert_mem_to_bytes(s):
     if s.endswith("G"):
-        return int(s[:-1]) * 10**6
+        return int(s[:-1]) * 10**9
     else:
         return int(s)
diff --git a/tests/tests/test_slurmio.py b/tests/tests/test_slurmio.py
@@ -20,8 +20,8 @@ def test_slurm_environment():
     assert 12 == params.allocated_cores
     assert 1 == params.requested_nodes
     assert 1 == params.allocated_nodes
-    assert 18000000 == params.requested_memory
-    assert 18000000 == params.allocated_memory
+    assert 18 * 1e9 == params.requested_memory
+    assert 18 * 1e9 == params.allocated_memory
     assert "12" == params._trackable_resources["cpu"]
     assert "18G" == params._trackable_resources["mem"]
     assert "1" == params._trackable_resources["node"]