pep8 compliant

Author: msbrogli

.flake8

@@ -0,0 +1,3 @@
+[flake8]
+ignore=E731
+max-line-length=160

sdm/__init__.py

@@ -23,7 +23,7 @@ def get_config_var(name):
     else:
         ext = get_config_var('SO')
 
-    fullpath = os.path.join(basedir, '_libsdm'+ext)
+    fullpath = os.path.join(basedir, '_libsdm' + ext)
     if not os.path.isfile(fullpath):
         fullpath = os.path.join(basedir, '_libsdm.so')
 
@@ -65,7 +65,7 @@ def _multK(value, K):
 def opencl_worksize_mult16(address_space):
     local_worksize = _multK(address_space.bs_len, 16)
     mcu = address_space.opencl_opts.max_compute_units
-    global_worksize = _multK(address_space.sample // 20, 2*mcu*local_worksize)
+    global_worksize = _multK(address_space.sample // 20, 2 * mcu * local_worksize)
     return local_worksize, global_worksize
 
 
@@ -74,7 +74,7 @@ def opencl_worksize_power2(address_space):
     while local_worksize < address_space.bs_len:
         local_worksize *= 2
     mcu = address_space.opencl_opts.max_compute_units
-    global_worksize = _multK(address_space.sample // 20, 2*mcu*local_worksize)
+    global_worksize = _multK(address_space.sample // 20, 2 * mcu * local_worksize)
     return local_worksize, global_worksize
 
 
@@ -563,19 +563,19 @@ def write(self, addr, datum, radius=None, weight=1):
             if isinstance(weight, int):
                 libsdm.sdm_write2_weighted(pointer(self), addr.bs_data, c_uint(radius), datum.bs_data, c_int(weight))
             elif isinstance(weight, (list, tuple)):
-                assert(self.bits+1 == len(weight))
+                assert(self.bits + 1 == len(weight))
                 # See https://docs.python.org/3/library/ctypes.html#type-conversions
-                weight_buf = (c_int * (self.bits+1))(*weight)
+                weight_buf = (c_int * (self.bits + 1))(*weight)
                 libsdm.sdm_write2_weighted_table(pointer(self), addr.bs_data, c_uint(radius), datum.bs_data, weight_buf)
             else:
                 raise NotImplemented
 
-    #def write_sub(self, addr, datum, radius=None):
-    #    ''' Write a bitstring to the SDM.
-    #    '''
-    #    if radius is None:
-    #        radius = self.radius
-    #    libsdm.sdm_write_sub(pointer(self), addr.bs_data, c_uint(radius), datum.bs_data)
+    # def write_sub(self, addr, datum, radius=None):
+    #     ''' Write a bitstring to the SDM.
+    #     '''
+    #     if radius is None:
+    #         radius = self.radius
+    #     libsdm.sdm_write_sub(pointer(self), addr.bs_data, c_uint(radius), datum.bs_data)
 
     def write_random_bitstrings(self, n):
         for _ in range(n):

sdm/utils.py

@@ -21,26 +21,26 @@ def div_pow2(x, a):
     try:
         r = x
         for _ in range(a):
-            r = r/2.0
+            r = r / 2.0
         return r
     except OverflowError:
-        return 2**(log(x)/log(2) - a)
+        return 2**(log(x) / log(2) - a)
 
 
 def calculate_probabilities(bits):
     from math import factorial
-    comb = lambda a, b: factorial(a)//factorial(b)//factorial(a-b)
+    comb = lambda a, b: factorial(a) // factorial(b) // factorial(a - b)
     acc = [0]
-    for i in range(bits+1):
+    for i in range(bits + 1):
         acc.append(acc[-1] + comb(bits, i))
     return [div_pow2(x, bits) for x in acc[1:]]
 
 
 def __calculate_radius(bits, threshold=0.001):
     from math import factorial
-    comb = lambda a, b: factorial(a)//factorial(b)//factorial(a-b)
+    comb = lambda a, b: factorial(a) // factorial(b) // factorial(a - b)
     x = 0
-    for i in range(bits+1):
+    for i in range(bits + 1):
         x += comb(bits, i)
         p = div_pow2(x, bits)
         if p >= threshold:
@@ -50,11 +50,11 @@ def __calculate_radius(bits, threshold=0.001):
 
 def calculate_radius(bits, threshold=0.001):
     from scipy.stats import norm
-    return int(bits/2.0 + norm.ppf(threshold)*(bits**0.5)/2.0)
+    return int(bits / 2.0 + norm.ppf(threshold) * (bits**0.5) / 2.0)
 
     if threshold == 0.001:
         # norminv(0.001) = 3.090232306167814
-        return int(bits/2.0 - 3.090232306167814*(bits**0.5)/2.0)
+        return int(bits / 2.0 - 3.090232306167814 * (bits**0.5) / 2.0)
     return __calculate_radius(bits, threshold)