TST: fix up issues in requires_memory decorator

Fix wrong multiplier for /proc/meminfo, and do style cleanups.

1 files changed, 6 insertions, 5 deletions

diff --git a/numpy/testing/_private/utils.py b/numpy/testing/_private/utils.py
index 4642cc0f8..32bd738bc 100644
--- a/numpy/testing/_private/utils.py
+++ b/numpy/testing/_private/utils.py
@@ -2417,9 +2417,10 @@ def requires_memory(free_bytes):
 
 def _parse_size(size_str):
     """Convert memory size strings ('12 GB' etc.) to float"""
-    suffixes = {'': 1.0, 'b': 1.0,
-                'k': 1e3, 'm': 1e6, 'g': 1e9, 't': 1e12,
-                'kb': 1e3, 'mb': 1e6, 'gb': 1e9, 'tb': 1e12}
+    suffixes = {'': 1, 'b': 1,
+                'k': 1000, 'm': 1000**2, 'g': 1000**3, 't': 1000**4,
+                'kb': 1000, 'mb': 1000**2, 'gb': 1000**3, 'tb': 1000**4,
+                'kib': 1024, 'mib': 1024**2, 'gib': 1024**3, 'tib': 1024**4}
 
     size_re = re.compile(r'^\s*(\d+|\d+\.\d+)\s*({0})\s*$'.format(
         '|'.join(suffixes.keys())), re.I)
@@ -2427,7 +2428,7 @@ def _parse_size(size_str):
     m = size_re.match(size_str.lower())
     if not m or m.group(2) not in suffixes:
         raise ValueError("value {!r} not a valid size".format(size_str))
-    return float(m.group(1)) * suffixes[m.group(2)]
+    return int(float(m.group(1)) * suffixes[m.group(2)])
 
 
 def _get_mem_available():
@@ -2443,7 +2444,7 @@ def _get_mem_available():
         with open('/proc/meminfo', 'r') as f:
             for line in f:
                 p = line.split()
-                info[p[0].strip(':').lower()] = float(p[1]) * 1e3
+                info[p[0].strip(':').lower()] = int(p[1]) * 1024
 
         if 'memavailable' in info:
             # Linux >= 3.14