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
|
from __future__ import annotations
import shutil
import subprocess
def test(dials_data, tmp_path):
experiments = dials_data("centroid_test_data", pathlib=True) / "experiments.json"
result = subprocess.run(
[
shutil.which("dials.background"),
"output.plot=background.png",
"image=1",
experiments,
],
cwd=tmp_path,
capture_output=True,
)
assert not result.returncode and not result.stderr
assert (tmp_path / "background.png").is_file()
for line in result.stdout.splitlines():
if line.startswith(b"Mean background"):
assert line.endswith(b"0.559")
break
def test_checkpoints(dials_data, tmp_path):
experiments = dials_data("centroid_test_data", pathlib=True) / "experiments.json"
result = subprocess.run(
[
shutil.which("dials.background"),
"n_checkpoints=3",
experiments,
],
cwd=tmp_path,
capture_output=True,
)
assert not result.returncode and not result.stderr
images = [
line.decode()
for line in result.stdout.splitlines()
if b"For imageset 0" in line
]
images = [
line.replace("For imageset 0 image", "").replace(":", "") for line in images
]
images = {int(e) for e in images}
assert len(images) == 3
assert sorted(images) == [1, 5, 9]
def test_multiple_imagesets(dials_data, tmp_path):
filenames = sorted(
dials_data("thaumatin_grid_scan", pathlib=True).glob("thau_3_2_00*.cbf.bz2")
)
filenames.extend(
sorted(dials_data("centroid_test_data", pathlib=True).glob("centroid_*.cbf"))
)
result = subprocess.run(
[
shutil.which("dials.background"),
"output.plot=background.png",
"image=1,2",
"size_inches=16,10",
]
+ filenames,
cwd=tmp_path,
capture_output=True,
)
assert not result.returncode and not result.stderr
assert (tmp_path / "background.png").is_file()
lines = result.stdout.splitlines()
assert b"For imageset 0 image 1:" in lines
assert b"For imageset 0 image 2:" in lines
assert b"For imageset 1 image 1:" in lines
assert b"For imageset 1 image 2:" in lines
|
