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
|
# SPDX-License-Identifier: (Apache-2.0 OR MIT)
# Copyright ijl (2018-2025)
import lzma
import os
import sys
import sysconfig
from pathlib import Path
from typing import Any
IS_FREETHREADING = sysconfig.get_config_var("Py_GIL_DISABLED")
SUPPORTS_MEMORYVIEW = sys.implementation == "cpython"
SUPPORTS_GETREFCOUNT = sys.implementation == "cpython"
numpy = None # type: ignore
if not IS_FREETHREADING:
try:
import numpy # type: ignore # noqa: F401
except ImportError:
pass
pandas = None # type: ignore
if not IS_FREETHREADING:
try:
import pandas # type: ignore # noqa: F401
except ImportError:
pass
import pytest
import orjson
data_dir = os.path.join(os.path.dirname(__file__), "../data")
STR_CACHE: dict[str, str] = {}
OBJ_CACHE: dict[str, Any] = {}
def read_fixture_bytes(filename, subdir=None):
if subdir is None:
path = Path(data_dir, filename)
else:
path = Path(data_dir, subdir, filename)
if path.suffix == ".xz":
contents = lzma.decompress(path.read_bytes())
else:
contents = path.read_bytes()
return contents
def read_fixture_str(filename, subdir=None):
if filename not in STR_CACHE:
STR_CACHE[filename] = read_fixture_bytes(filename, subdir).decode("utf-8")
return STR_CACHE[filename]
def read_fixture_obj(filename):
if filename not in OBJ_CACHE:
OBJ_CACHE[filename] = orjson.loads(read_fixture_str(filename))
return OBJ_CACHE[filename]
needs_data = pytest.mark.skipif(
not Path(data_dir).exists(),
reason="Test depends on ./data dir that contains fixtures",
)
|
