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
|
import pytest
import fiona
from fiona.model import Feature, Geometry, Properties
class TestReadAccess:
# To check that we'll be able to get multiple 'r' connections to layers
# in a single file.
def test_meta(self, path_coutwildrnp_shp):
with fiona.open(path_coutwildrnp_shp, "r", layer="coutwildrnp") as c:
with fiona.open(path_coutwildrnp_shp, "r", layer="coutwildrnp") as c2:
assert len(c) == len(c2)
assert sorted(c.schema.items()) == sorted(c2.schema.items())
def test_feat(self, path_coutwildrnp_shp):
with fiona.open(path_coutwildrnp_shp, "r", layer="coutwildrnp") as c:
f1 = next(iter(c))
with fiona.open(path_coutwildrnp_shp, "r", layer="coutwildrnp") as c2:
f2 = next(iter(c2))
assert f1.id == f2.id
assert f1.properties == f2.properties
assert f1.geometry.type == f2.geometry.type
class TestReadWriteAccess:
# To check that we'll be able to read from a file that we're
# writing to.
@pytest.fixture(autouse=True)
def multi_write_test_shp(self, tmpdir):
self.shapefile_path = str(tmpdir.join("multi_write_test.shp"))
self.c = fiona.open(
self.shapefile_path,
"w",
driver="ESRI Shapefile",
schema={
"geometry": "Point",
"properties": [("title", "str:80"), ("date", "date")],
},
crs={"init": "epsg:4326", "no_defs": True},
encoding="utf-8",
)
self.f = Feature(
id="0",
geometry=Geometry(type="Point", coordinates=(0.0, 0.1)),
properties=Properties(title="point one", date="2012-01-29"),
)
self.c.writerecords([self.f])
self.c.flush()
yield
self.c.close()
def test_meta(self):
c2 = fiona.open(self.shapefile_path, "r")
assert len(self.c) == len(c2)
assert sorted(self.c.schema.items()) == sorted(c2.schema.items())
c2.close()
def test_read(self):
c2 = fiona.open(self.shapefile_path, "r")
f2 = next(iter(c2))
assert self.f.id == f2.id
assert self.f.properties == f2.properties
assert self.f.geometry.type == f2.geometry.type
c2.close()
def test_read_after_close(self):
c2 = fiona.open(self.shapefile_path, "r")
self.c.close()
f2 = next(iter(c2))
assert self.f.properties == f2.properties
c2.close()
class TestLayerCreation:
@pytest.fixture(autouse=True)
def layer_creation_shp(self, tmpdir):
self.dir = tmpdir.mkdir("layer_creation")
self.c = fiona.open(
str(self.dir),
"w",
layer="write_test",
driver="ESRI Shapefile",
schema={
"geometry": "Point",
"properties": [("title", "str:80"), ("date", "date")],
},
crs={"init": "epsg:4326", "no_defs": True},
encoding="utf-8",
)
self.f = Feature(
geometry=Geometry(type="Point", coordinates=(0.0, 0.1)),
properties={"title": "point one", "date": "2012-01-29"},
)
self.c.writerecords([self.f])
self.c.flush()
yield
self.c.close()
def test_meta(self):
c2 = fiona.open(str(self.dir.join("write_test.shp")), "r")
assert len(self.c) == len(c2)
assert sorted(self.c.schema.items()) == sorted(c2.schema.items())
c2.close()
def test_read(self):
c2 = fiona.open(str(self.dir.join("write_test.shp")), "r")
f2 = next(iter(c2))
assert self.f.properties == f2.properties
c2.close()
def test_read_after_close(self):
c2 = fiona.open(str(self.dir.join("write_test.shp")), "r")
self.c.close()
f2 = next(iter(c2))
assert self.f.properties == f2.properties
c2.close()
|
