/usr/lib/python3/dist-packages/joblib/test/test_disk.py is in python3-joblib 0.10.3+git55-g660fe5d-1.
This file is owned by root:root, with mode 0o644.
The actual contents of the file can be viewed below.
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 | """
Unit tests for the disk utilities.
"""
# Authors: Gael Varoquaux <gael dot varoquaux at normalesup dot org>
# Lars Buitinck
# Copyright (c) 2010 Gael Varoquaux
# License: BSD Style, 3 clauses.
from __future__ import with_statement
import os
import shutil
import array
from tempfile import mkdtemp
from joblib.disk import disk_used, memstr_to_bytes, mkdirp
from joblib.testing import assert_equal, assert_raises
###############################################################################
def test_disk_used():
cachedir = mkdtemp()
try:
if os.path.exists(cachedir):
shutil.rmtree(cachedir)
os.mkdir(cachedir)
# Not write a file that is 1M big in this directory, and check the
# size. The reason we use such a big file is that it makes us robust
# to errors due to block allocation.
a = array.array('i')
sizeof_i = a.itemsize
target_size = 1024
n = int(target_size * 1024 / sizeof_i)
a = array.array('i', n * (1,))
with open(os.path.join(cachedir, 'test'), 'wb') as output:
a.tofile(output)
assert disk_used(cachedir) >= target_size
assert disk_used(cachedir) < target_size + 12
finally:
shutil.rmtree(cachedir)
def test_memstr_to_bytes():
for text, value in zip(('80G', '1.4M', '120M', '53K'),
(80 * 1024 ** 3, int(1.4 * 1024 ** 2),
120 * 1024 ** 2, 53 * 1024)):
yield assert_equal, memstr_to_bytes(text), value
assert_raises(ValueError, memstr_to_bytes, 'foobar')
def test_mkdirp():
try:
tmp = mkdtemp()
mkdirp(os.path.join(tmp, "ham"))
mkdirp(os.path.join(tmp, "ham"))
mkdirp(os.path.join(tmp, "spam", "spam"))
# Not all OSErrors are ignored
assert_raises(OSError, mkdirp, "")
finally:
shutil.rmtree(tmp)
|