blob: 399f9f76516e53b9f4af15568f8100d6455241cd (
plain)
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
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
|
import decimal
import pickle
import time
import pytest
import pint
from pint._vendor import flexparser as fp
from pint.facets.plain import UnitDefinition
FS_SLEEP = 0.010
@pytest.fixture
def float_cache_filename(tmp_path):
ureg = pint.UnitRegistry(cache_folder=tmp_path / "cache_with_float")
assert ureg._diskcache
assert ureg._diskcache.cache_folder
return tuple(ureg._diskcache.cache_folder.glob("*.pickle"))
def test_must_be_three_files(float_cache_filename):
# There should be 3 files here:
# - cached defaults_en.txt
# - cached constants_en.txt
# - cached RegistryCache
assert len(float_cache_filename) == 3, float_cache_filename
def test_no_cache():
ureg = pint.UnitRegistry(cache_folder=None)
assert ureg._diskcache is None
assert ureg.cache_folder is None
def test_decimal(tmp_path, float_cache_filename):
ureg = pint.UnitRegistry(
cache_folder=tmp_path / "cache_with_decimal", non_int_type=decimal.Decimal
)
assert ureg._diskcache
assert ureg._diskcache.cache_folder == tmp_path / "cache_with_decimal"
assert ureg.cache_folder == tmp_path / "cache_with_decimal"
files = tuple(ureg._diskcache.cache_folder.glob("*.pickle"))
assert len(files) == 3
# check that the filenames with decimal are different to the ones with float
float_filenames = tuple(p.name for p in float_cache_filename)
for p in files:
assert p.name not in float_filenames
for p in files:
with p.open(mode="rb") as fi:
obj = pickle.load(fi)
if not isinstance(obj, fp.ParsedSource):
continue
for definition in obj.parsed_source.filter_by(UnitDefinition):
if definition.name == "pi":
assert isinstance(definition.converter.scale, decimal.Decimal)
return
assert False
def test_auto(float_cache_filename):
float_filenames = tuple(p.name for p in float_cache_filename)
ureg = pint.UnitRegistry(cache_folder=":auto:")
assert ureg._diskcache
assert ureg._diskcache.cache_folder
auto_files = tuple(p.name for p in ureg._diskcache.cache_folder.glob("*.pickle"))
for file in float_filenames:
assert file in auto_files
def test_change_file(tmp_path):
# Generate a definition file
dfile = tmp_path / "definitions.txt"
dfile.write_text("x = 1234")
# Load the definition file
# (this will create two cache files, one for the file another for RegistryCache)
ureg = pint.UnitRegistry(dfile, cache_folder=tmp_path)
assert ureg.x == 1234
files = tuple(ureg._diskcache.cache_folder.glob("*.pickle"))
assert len(files) == 2
# Modify the definition file
# Add some sleep to make sure that the time stamp difference is signficant.
time.sleep(FS_SLEEP)
dfile.write_text("x = 1235")
# Verify that the definiton file was loaded (the cache was invalidated).
ureg = pint.UnitRegistry(dfile, cache_folder=tmp_path)
assert ureg.x == 1235
files = tuple(ureg._diskcache.cache_folder.glob("*.pickle"))
assert len(files) == 4
|
