testing mlock

examples/mlock.py

@@ -1,5 +1,4 @@
 import numpy as np
-import array
 import platform
 import random
 import os
@@ -56,23 +55,17 @@ def setup_memory_limit():
         arr[:] = os.urandom(len(arr))
         arr_np = np.zeros(len(arr), dtype=np.uint8)
         arr_np[:] = arr
-        array_array = array.array('B', (random.randint(0, 255) for _ in range(int(size * 1024 * 1024))))
         print(f"Testing size: {size} MB")
         print("mlock bytearray")
         mlock(arr)
         print("mlock np array")
         mlock(arr_np)
-        print("mlock array.array")
-        mlock(array_array)
 
         zeroize1(arr)
         zeroize1(arr_np)
-        zeroize1(array_array)
         assert arr == bytearray(int(size * 1024 * 1024))
         assert all(arr_np == 0)
-        assert all(byte == 0 for byte in array_array)
 
     finally:
         munlock(arr)
         munlock(arr_np)
-        munlock(array_array)

tests/test_zeroize.py

@@ -80,26 +80,20 @@ def test_zeroize1_sizes(self):
                 arr[:] = os.urandom(len(arr))
                 arr_np = np.zeros(len(arr), dtype=np.uint8)
                 arr_np[:] = arr
-                array_array = array.array('B', (random.randint(0, 255) for _ in range(int(size * 1024 * 1024))))
                 print(f"Testing size: {size} MB")
                 print("mlock bytearray")
                 mlock(arr)
                 print("mlock np array")
                 mlock(arr_np)
-                print("mlock array.array")
-                mlock(array_array)
 
                 zeroize1(arr)
                 zeroize1(arr_np)
-                zeroize1(array_array)
                 self.assertEqual(arr, bytearray(int(size * 1024 * 1024)))
                 self.assertEqual(True, all(arr_np == 0))
-                self.assertEqual(True, all(byte == 0 for byte in array_array))
 
             finally:
                 munlock(arr)
                 munlock(arr_np)
-                munlock(array_array)
 
 
 if __name__ == "__main__":