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
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
|
import io
import numpy as np
import pytest
import laspy
from laspy import PointFormat
from tests.test_common import simple_las, test1_4_las, write_then_read_again
@pytest.fixture()
def file1_4():
return laspy.read(test1_4_las)
@pytest.fixture()
def file():
return laspy.read(simple_las)
def test_xyz():
las = laspy.create()
shape = (150,)
las.X = np.zeros(shape, dtype=np.int32)
las.Y = np.ones(shape, dtype=np.int32)
las.Z = np.zeros(shape, dtype=np.int32)
las.Z[:] = -152
las = write_then_read_again(las)
assert np.all(las.X == 0)
assert np.all(las.Y == 1)
assert np.all(las.Z == -152)
def test_wrong_version():
for i in range(6, 8):
with pytest.raises(laspy.errors.LaspyException):
_ = laspy.create(point_format=i, file_version="1.2")
def test_good_version_is_used():
for i in range(6, 8):
las = laspy.create(point_format=i)
assert las.header.version.major == 1
assert las.header.version.minor == 4
def test_create_fmt_0():
new = laspy.create(point_format=0)
with pytest.raises(ValueError):
new.red = np.zeros(len(new.points), np.uint16)
with pytest.raises(ValueError):
new.red = np.zeros(len(new.points), np.uint16)
with pytest.raises(ValueError):
new.red = np.zeros(len(new.points), np.uint16)
with pytest.raises(ValueError):
new.gps_time = np.zeros(len(new.points), np.float64)
def test_create_fmt_1():
new = laspy.create(point_format=1)
with pytest.raises(ValueError):
new.red = np.zeros(len(new.points), np.uint16)
with pytest.raises(ValueError):
new.red = np.zeros(len(new.points), np.uint16)
with pytest.raises(ValueError):
new.red = np.zeros(len(new.points), np.uint16)
gps_time = np.random.uniform(0, 25641, len(new.points))
new.gps_time = gps_time
assert np.allclose(new.gps_time, gps_time)
new = write_then_read_again(new)
assert np.allclose(new.gps_time, gps_time)
def test_create_fmt_2(file):
new = laspy.create(point_format=2)
with pytest.raises(ValueError):
new.gps_time = file.gps_time
new.red = file.red
new.green = file.green
new.blue = file.blue
assert np.allclose(new.red, file.red)
assert np.allclose(new.green, file.green)
assert np.allclose(new.blue, file.blue)
new = write_then_read_again(new)
assert np.allclose(new.red, file.red)
assert np.allclose(new.green, file.green)
assert np.allclose(new.blue, file.blue)
def test_create_fmt_3(file):
new = laspy.create(point_format=3)
new.red = file.red
new.green = file.green
new.blue = file.blue
new.gps_time = file.gps_time
assert np.allclose(new.red, file.red)
assert np.allclose(new.green, file.green)
assert np.allclose(new.blue, file.blue)
assert np.allclose(new.gps_time, file.gps_time)
new = write_then_read_again(new)
assert np.allclose(new.red, file.red)
assert np.allclose(new.green, file.green)
assert np.allclose(new.blue, file.blue)
assert np.allclose(new.gps_time, file.gps_time)
def test_create_fmt_6(file1_4):
new = laspy.create(point_format=6)
assert str(new.header.version) == "1.4"
dim_names_fmt_6 = PointFormat(6).dtype().names
for dim_name in dim_names_fmt_6:
new[dim_name] = file1_4[dim_name]
for dim_name in dim_names_fmt_6:
assert np.allclose(new[dim_name], file1_4[dim_name]), "{} not equal".format(
dim_name
)
new = write_then_read_again(new)
for dim_name in dim_names_fmt_6:
assert np.allclose(new[dim_name], file1_4[dim_name]), "{} not equal".format(
dim_name
)
@pytest.mark.parametrize("laz_backend", (None,) + laspy.LazBackend.detect_available())
def test_writing_empty_file(laz_backend):
las = laspy.create()
with io.BytesIO() as out:
if laz_backend is None:
las.write(out)
else:
las.write(out, laz_backend=laz_backend)
def test_changing_scales_offset_after_create():
las = laspy.create(point_format=8, file_version="1.4")
las.header.x_scale = 0.0001
las.header.y_scale = 0.0001
las.header.z_scale = 0.0001
las.x = np.ones((10,)) * 11
las.y = np.ones((10,)) * 12
las.z = np.ones((10,)) * 13
las.x[0] = 1
las.y[0] = 2
las.z[0] = 3
las.update_header()
assert np.all(las.header.mins == [1, 2, 3])
assert np.all(las.header.maxs == [11, 12, 13])
assert las.x.min() == 1
assert las.y.min() == 2
assert las.z.min() == 3
assert las.x.max() == 11
assert las.y.max() == 12
assert las.z.max() == 13
|
