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
|
# SPDX-License-Identifier: BSD-2-Clause
#
# This file is part of pyosmium. (https://osmcode.org/pyosmium/)
#
# Copyright (C) 2025 Sarah Hoffmann <lonvia@denofr.de> and others.
# For a full list of authors see the git log.
import pytest
import osmium
def test_read_node_location_with_handler(opl_reader):
idx = osmium.index.create_map("flex_mem")
hdlr = osmium.NodeLocationsForWays(idx)
data = """\
n1 x6 y7
n45 x-3 y0
"""
osmium.apply(opl_reader(data), hdlr)
assert idx.get(1).lon == pytest.approx(6)
assert idx.get(1).lat == pytest.approx(7)
assert idx.get(45).lon == pytest.approx(-3)
assert idx.get(45).lat == 0.0
with pytest.raises(KeyError):
idx.get(2)
@pytest.mark.parametrize('ignore_error', [True, False])
def test_apply_node_location_handler(opl_reader, ignore_error):
hdlr = osmium.NodeLocationsForWays(osmium.index.create_map("flex_mem"))
if ignore_error:
hdlr.ignore_errors()
class WayNodeHandler:
def __init__(self):
self.collect = []
self.with_error = []
def way(self, w):
try:
self.collect.append((w.id, [(n.lon, n.lat) for n in w.nodes]))
except osmium.InvalidLocationError:
self.with_error.append(w.id)
data = """\
n1 x6 y7
n2 x6 y7.1
n45 x-3 y0
n55 x-2.9 y0
w3 Nn1,n2
w4 Nn45,n55,n56
"""
tester = WayNodeHandler()
if ignore_error:
osmium.apply(opl_reader(data), hdlr, tester)
assert tester.collect == [(3, [(pytest.approx(6), pytest.approx(7)),
(pytest.approx(6), pytest.approx(7.1))])]
assert tester.with_error == [4]
else:
with pytest.raises(KeyError):
osmium.apply(opl_reader(data), hdlr, tester)
def test_apply_invalid_handler_object(opl_reader):
class DummyHandler:
def some_func():
print('A')
with pytest.raises(TypeError):
osmium.apply(opl_reader("n1 x2 z4"), DummyHandler())
def test_mixed_handlers(opl_reader):
logged = []
class OldStyle(osmium.SimpleHandler):
def node(self, n):
logged.append('old')
class NewStyle:
def node(self, n):
logged.append('new')
osmium.apply(opl_reader("n1 x0 y0"), NewStyle(), OldStyle(), NewStyle(), OldStyle())
assert logged == ['new', 'old', 'new', 'old']
def test_value_propagation(opl_reader):
logged = []
class FirstHandler:
def node(self, n):
n.saved = 45674
class SecondHandler:
def node(self, n):
logged.append(n.saved)
osmium.apply(opl_reader("n1 x0 y0"), FirstHandler(), SecondHandler())
assert logged == [45674]
|
